結果
問題 | No.2944 Sigma Partition Problem |
ユーザー |
|
提出日時 | 2024-10-18 22:33:19 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 171 ms / 4,000 ms |
コード長 | 826 bytes |
コンパイル時間 | 5,517 ms |
コンパイル使用メモリ | 311,060 KB |
実行使用メモリ | 191,996 KB |
最終ジャッジ日時 | 2024-10-18 22:52:29 |
合計ジャッジ時間 | 10,049 ms |
ジャッジサーバーID (参考情報) |
judge7 / judge |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 24 |
ソースコード
#include<bits/stdc++.h> using namespace std; #include<atcoder/all> using namespace atcoder; using mint=atcoder::modint998244353; #pragma GCC optimize("O3") #pragma GCC optimize("unroll-loops") #define int long long mint dp1[4009][4009]; mint dp2[4009][4009]; mint dp[4009][4009]; signed main(){ dp1[0][0]=1; for(int i=1;i<=4000;i++){ for(int j=0;j<=4000;j++)dp1[i][j]=dp1[i-1][j]; for(int j=0;i+j<=4000;j++)dp1[i][j+i]+=dp1[i][j]; } for(int i=1;i<=4000;i++){ for(int j=0;j<=4000;j++){ for(int k=1;i*k+j<=4000;k++){ dp2[i][i*k+j]+=dp1[i-1][j]; } } } for(int i=0;i<=4000;i++){ for(int j=0;j<=4000;j++){ dp2[i+1][j]+=dp2[i][j]; } } int Q;cin>>Q; while(Q--){ int t,n,k;cin>>t>>n>>k; if(t==1){ cout<<dp1[k][n].val()<<endl; }else{ cout<<dp2[k][n].val()<<endl; } } }