結果

問題 No.2874 Gunegune Tree
ユーザー detteiuu
提出日時 2024-12-01 02:02:16
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 157 ms / 2,000 ms
コード長 959 bytes
コンパイル時間 236 ms
コンパイル使用メモリ 82,104 KB
実行使用メモリ 99,864 KB
最終ジャッジ日時 2024-12-01 02:02:22
合計ジャッジ時間 4,889 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

N = int(input())

def inverse(n, d):
    return n * pow(d, -1, MOD) % MOD

MOD = 998244353

P = [0, 1, inverse(1, 2), inverse(1, 3)]
dp = [[0]*5 for _ in range(N)]
for i in range(1, N):
    for j in range(5):
        for k in range(-1, 2):
            if 0 <= j+k <= 4:
                if 1 <= j+k <= 3:
                    if 1 <= j <= 3:
                        dp[i][j] += (dp[i-1][j+k]+1)*P[3]%MOD
                    else:
                        dp[i][j] += (dp[i-1][j+k]+1)*P[2]%MOD
                    dp[i][j] %= MOD
                else:
                    if 1 <= j <= 3:
                        dp[i][j] += dp[i-1][j+k]*P[3]%MOD
                    else:
                        dp[i][j] += dp[i-1][j+k]*P[2]%MOD
                    dp[i][j] %= MOD

ans = 0
for i in range(5):
    if 1 <= i <= 3:
        ans += (dp[-1][i]+1)*inverse(1, 5)%MOD
        ans %= MOD
    else:
        ans += dp[-1][i]*inverse(1, 5)%MOD
        ans %= MOD

print(ans)
0