結果
問題 | No.1549 [Cherry 2nd Tune] BANning Tuple |
ユーザー |
👑 ![]() |
提出日時 | 2021-06-08 13:29:59 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 1,102 ms / 4,000 ms |
コード長 | 2,710 bytes |
コンパイル時間 | 1,807 ms |
コンパイル使用メモリ | 95,668 KB |
最終ジャッジ日時 | 2025-01-22 04:55:34 |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 20 |
ソースコード
#include <iostream> #include <vector> #include <map> #include <algorithm> using namespace std; using ll = long long; using ull = unsigned long long; const ull M = 998244353; ull powm(ull a, ull i) { if(i==0) return 1; ull r=powm(a*a%M,i/2); if(i&1) r=r*a%M; return r; } ull invm(ull a){ return powm(a,M-2); } void NTT(vector<ull>& A, ull g){ int N=A.size(); for(int i=0,j=0; j<N; j++){ if(i<j) swap(A[i],A[j]); for(int k=N>>1; k>(i^=k); k>>=1); } for(int i=1; i<N; i<<=1){ ull q=powm(g,(M-1)/i/2), qj=1; for(int j=0; j<i; j++){ for(int k=j; k<N; k+=i*2){ ull l=A[k],r=A[k+i]*qj%M; A[k]=(l+r); if(A[k]>=M) A[k]-=M; A[k+i]=(l+M-r); if(A[k+i]>=M) A[k+i]-=M; } qj=qj*q%M; } } } const int maxT = 3000; const int NTTZ = 1<<13; const ull NTTg = 3; const ull NTTinvZ = invm(NTTZ); const ull NTTinvg = invm(NTTg); vector<ull> convolute(vector<ull> l, vector<ull> r){ NTT(l,NTTg); NTT(r,NTTg); vector<ull> ans(NTTZ,0); for(int i=0; i<NTTZ; i++) ans[i] = l[i] * r[i] % M * NTTinvZ % M; NTT(ans,NTTinvg); for(int i=maxT+1; i<NTTZ; i++) ans[i] = 0; return move(ans); } ull N; int Q; map<ull,int> Idx; vector<vector<ull>> VecList; vector<ull> AnsArr; const int segtreeN = 100; vector<vector<ull>> segtree; void segtree_upd(int p, vector<ull> v){ p += segtreeN; segtree[p] = move(v); while(p != 1){ p /= 2; segtree[p] = convolute(segtree[p*2],segtree[p*2+1]); } } int main(){ cin>>N>>Q; segtree.assign(segtreeN*2, vector<ull>(NTTZ,0)); for(auto& p : segtree) p[0] = 1; for(int q=0; q<Q; q++){ ull K; cin>>K; int A,B,S,T; cin>>A>>B>>S>>T; if(Idx.count(K) == 0){ Idx[K] = VecList.size(); VecList.push_back(vector<ull>(NTTZ,0)); for(int i=0; i<=3000; i++) VecList.back()[i] = 1; } int idx = Idx[K]; for(int x=A; x<=B; x++) VecList[idx][x] = 0; segtree_upd(idx,VecList[idx]); AnsArr = segtree[1]; ull leftN = N - VecList.size(); vector<ull> ansCoeff(NTTZ+1); ansCoeff[0] = 1; for(int i=1; i<=NTTZ; i++) ansCoeff[i] = ansCoeff[i-1] * ((leftN+i-1) % M) % M * invm(i) % M; reverse(ansCoeff.begin(),ansCoeff.end()); ansCoeff.push_back(0); reverse(ansCoeff.begin(),ansCoeff.end()); for(int i=0; i<=NTTZ; i++) ansCoeff[i+1] = (ansCoeff[i] + ansCoeff[i+1]) % M; ull ans = 0; for(int s=0; s<NTTZ; s++){ ull coeff = ansCoeff[max(0,T-s+1)] + M - ansCoeff[max(0,S-s)]; ans = (ans + AnsArr[s] * coeff) % M; } cout << (ans % M) << "\n"; } return 0; } struct ios_do_not_sync{ ios_do_not_sync(){ ios::sync_with_stdio(false); cin.tie(nullptr); } } ios_do_not_sync_instance;