結果

問題 No.3579 区間積逆像
コンテスト
ユーザー tau1235
提出日時 2026-07-03 21:52:37
言語 C++23
(gcc 15.2.0 + boost 1.89.0)
コンパイル:
g++-15 -O2 -lm -std=c++23 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
WA  
実行時間 -
コード長 886 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,387 ms
コンパイル使用メモリ 348,316 KB
実行使用メモリ 14,700 KB
最終ジャッジ日時 2026-07-03 21:52:43
合計ジャッジ時間 5,264 ms
ジャッジサーバーID
（参考情報） 		judge2_0 / judge1_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 28 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include<bits/stdc++.h>
#include<atcoder/modint>
using namespace std;

int main(){
  using mint=atcoder::modint;
  int n,p,c;
  cin>>n>>p>>c;
  mint::set_mod(p);
  vector<mint> a(n);
  vector<vector<mint>> g;
  vector<mint> vec;
  for (int i=0;i<n;i++){
    int x;
    cin>>x;
    a[i]=x;
    if (x==0&&!vec.empty()) g.push_back(vec),vec.clear();
    if (x) vec.push_back(x);
  }
  if (!vec.empty()) g.push_back(vec);
  long long ans=0;
  if (c==0){
    ans=(long long)(n+1)*n/2;
    for (auto vec:g) ans-=((long long)(vec.size())+1)*(int)vec.size()/2;
  }
  else{
    mint cinv=mint(c).inv();
    auto f=[&](vector<mint> b){
      int m=b.size();
      map<int,int> mp;
      mint prod=1;
      mp[1%p]=1;
      for (int i=0;i<m;i++){
        prod*=a[i];
        ans+=mp[(prod*cinv).val()];
        mp[prod.val()]++;
      }
    };
    for (auto vec:g) f(vec);
  }
  cout<<ans<<endl;
}
0