結果
問題 |
No.3096 Snake Path
|
ユーザー |
|
提出日時 | 2025-04-06 16:16:54 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 22 ms / 2,000 ms |
コード長 | 1,313 bytes |
コンパイル時間 | 846 ms |
コンパイル使用メモリ | 66,352 KB |
実行使用メモリ | 7,844 KB |
最終ジャッジ日時 | 2025-04-06 16:16:56 |
合計ジャッジ時間 | 2,219 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 35 |
ソースコード
#include<iostream> #include<cassert> #include<atcoder/modint> using namespace std; using mint=atcoder::modint998244353; int N,K; mint dp[2][5][2020]; const int P[5]={0b001,0b010,0b100,0b111,0b111}; int main() { ios::sync_with_stdio(false); cin.tie(nullptr); cin>>N>>K; int now=0; dp[now][0][0]=1; for(int i=0;i<N;i++) { int nxt=1-now; for(int j=0;j<5;j++)for(int k=0;k<=K;k++)dp[nxt][j][k]=0; for(int k=0;k<=K;k++) { //0 dp[nxt][0][k]+=dp[now][0][k]; dp[nxt][1][k]+=dp[now][0][k]; dp[nxt][2][k]+=dp[now][0][k]; dp[nxt][3][k]+=dp[now][0][k]; //1 dp[nxt][0][k]+=dp[now][1][k]; dp[nxt][1][k]+=dp[now][1][k]; dp[nxt][2][k]+=dp[now][1][k]; //2 dp[nxt][0][k]+=dp[now][2][k]; dp[nxt][1][k]+=dp[now][2][k]; dp[nxt][2][k]+=dp[now][2][k]; dp[nxt][4][k]+=dp[now][2][k]; //3 dp[nxt][2][k]+=dp[now][3][k]; dp[nxt][3][k]+=dp[now][3][k]; //4 dp[nxt][0][k]+=dp[now][4][k]; dp[nxt][4][k]+=dp[now][4][k]; } now=nxt; int C[3]={1,1,1}; C[i%2==0?2:0]=0; for(int j=0;j<5;j++) { int c=0; for(int k=0;k<3;k++)if(P[j]>>k&1)c+=C[k]; for(int k=K;k>=0;k--) { mint t=dp[now][j][k]; dp[now][j][k]=0; dp[now][j][k+c]+=t; } } } mint ans=0; int j=N%2==1?2:0; for(int k=0;k<=K;k++)ans+=dp[now][j][k]; cout<<ans.val()<<endl; }