結果
問題 | No.1549 [Cherry 2nd Tune] BANning Tuple |
ユーザー |
|
提出日時 | 2021-06-11 23:59:47 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 2,578 ms / 4,000 ms |
コード長 | 1,493 bytes |
コンパイル時間 | 1,496 ms |
コンパイル使用メモリ | 96,916 KB |
実行使用メモリ | 6,992 KB |
最終ジャッジ日時 | 2024-12-15 04:04:45 |
合計ジャッジ時間 | 32,149 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 20 |
コンパイルメッセージ
main.cpp:11:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type] 11 | main() | ^~~~
ソースコード
#include<iostream> #include<map> #include<set> #include<vector> #include<atcoder/modint> using namespace std; using mint=atcoder::modint998244353; long N; int Q; mint C[101][3001]; main() { cin>>N>>Q; for(int i=0;i<=100;i++) { mint x=1,y=1; for(int j=0;j<=3000;j++) { C[i][j]=x/y; x*=N-i+j+1; y*=j+1; } } map<long,set<pair<int,int> > >mp; for(int q=0;q<Q;q++) { long K;int A,B,S,T; cin>>K>>A>>B>>S>>T; B++; { set<pair<int,int> >now; for(pair<int,int>p:mp[K]) { if(p.second<A)now.insert(p); else if(B<p.first)now.insert(p); else { if(A>p.first)A=p.first; if(B<p.second)B=p.second; } } now.insert(make_pair(A,B)); mp[K].swap(now); } vector<vector<mint> >dp(mp.size()+1,vector<mint>(T+1)); dp[0][0]=1; int cnt=0; for(auto&&S:mp) { vector<vector<mint> >ep(cnt+2,vector<mint>(T+1)); for(int i=cnt;i>=0;i--) { for(pair<int,int>p:S.second) { int a=p.first,b=p.second; for(int j=0;j+a<=T;j++) { ep[i+1][j+a]+=dp[i][j]; int r=j+b; if(r<=T)ep[i+1][r]-=dp[i][j]; } } } cnt++; for(int i=0;i<=cnt;i++) { for(int j=1;j<=T;j++)ep[i][j]+=ep[i][j-1]; for(int j=0;j<=T;j++)dp[i][j]+=ep[i][j]; } } mint ans=0; for(int i=0;i<=cnt;i++) { for(int j=0;j<=T;j++) { mint f=dp[i][j]; if(i&1)f=-f; mint val=C[i][T-j]; if(S>j) { val-=C[i][S-j-1]; } ans+=f*val; } } cout<<ans.val()<<endl; } }