結果

問題 No.2874 Gunegune Tree
ユーザー titia
提出日時 2024-09-11 03:04:58
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 1,261 ms / 2,000 ms
コード長 924 bytes
コンパイル時間 272 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 65,664 KB
最終ジャッジ日時 2024-09-11 03:05:19
合計ジャッジ時間 18,596 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline

mod=998244353

POW2=pow(2,mod-2,mod)
POW3=pow(3,mod-2,mod)
POW5=pow(5,mod-2,mod)

N=int(input())

DP=[[0,0,0,0,0] for i in range(N)]

DP[0][0]=POW5
DP[0][1]=POW5
DP[0][2]=POW5
DP[0][3]=POW5
DP[0][4]=POW5

for i in range(1,N):
    k=DP[i-1][0]

    DP[i][0]=(DP[i][0]+k*POW2)%mod
    DP[i][1]=(DP[i][1]+k*POW2)%mod

    k=DP[i-1][1]

    DP[i][0]=(DP[i][0]+k*POW3)%mod
    DP[i][1]=(DP[i][1]+k*POW3)%mod
    DP[i][2]=(DP[i][2]+k*POW3)%mod

    k=DP[i-1][2]

    DP[i][1]=(DP[i][1]+k*POW3)%mod
    DP[i][2]=(DP[i][2]+k*POW3)%mod
    DP[i][3]=(DP[i][3]+k*POW3)%mod

    k=DP[i-1][3]

    DP[i][2]=(DP[i][2]+k*POW3)%mod
    DP[i][3]=(DP[i][3]+k*POW3)%mod
    DP[i][4]=(DP[i][4]+k*POW3)%mod

    k=DP[i-1][4]

    DP[i][3]=(DP[i][3]+k*POW2)%mod
    DP[i][4]=(DP[i][4]+k*POW2)%mod

ANS=0

for i in range(N):
    for j in range(1,4):
        ANS=(ANS+DP[i][j])%mod

print(ANS)
0