結果

問題 No.3096 Snake Path
ユーザー autumn09
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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)
    
            
0