結果
問題 | No.3096 Snake Path |
ユーザー |
|
提出日時 | 2025-04-05 11:44:37 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,365 ms / 2,000 ms |
コード長 | 924 bytes |
コンパイル時間 | 454 ms |
コンパイル使用メモリ | 82,236 KB |
実行使用メモリ | 215,828 KB |
最終ジャッジ日時 | 2025-04-05 11:44:55 |
合計ジャッジ時間 | 17,482 ms |
ジャッジサーバーID (参考情報) |
judge6 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 35 |
ソースコード
MOD=998244353 N,K=map(int,input().split()) dp=[[[0]*5 for _ in range(K+1)] for _ in range(N)] dp[0][0][0]=1 def update(i,j,k, val): dp[i][j][k]+=val dp[i][j][k]%=MOD c=["z1o-2", "z1o--", "z1o2-", "z--2-", "--o-2"] def cost(f,t,b): if (c[f][t]=='-'): return 1e9 if (c[f][t]=='z'): return b if (c[f][t]=='o'): return (not b) return int(c[f][t]) for i in range(1,N): for j in range(K+1): for f in range(5): for t in range(5): nj=j+cost(f,t,i&1) if (nj>K): continue update(i,nj,t,dp[i-1][j][f]) ans=0 if (N&1): for k in range(K+1): for i in [0,1,2,4]: ans+=dp[N-1][k][i] ans%=MOD else: for k in range(K+1): for i in [0,1,2,3]: ans+=dp[N-1][k][i] ans%=MOD print(ans)