結果

問題 No.2874 Gunegune Tree
ユーザー ニックネーム
提出日時 2024-09-06 23:15:45
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 622 bytes
コンパイル時間 246 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 10,880 KB
最終ジャッジ日時 2024-09-06 23:17:18
合計ジャッジ時間 33,467 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24 TLE * 6
権限があれば一括ダウンロードができます

ソースコード

diff # 

n,z = int(input()),998244353
d,e,f = pow(5,-1,z),pow(3,-1,z),pow(2,-1,z)
p = [0,d,d,d,0]; q = [d]*5
for _ in range(n-1):
    pp = [0]*5; qq = [0]*5
    for i in range(5):
        if i==0:
            for j in range(2):
                pp[j] += (p[i]+(0<j)*q[i])*f
                qq[j] += q[i]*f
        elif i==4:
            for j in range(3,5):
                pp[j] += (p[i]+(j<4)*q[i])*f
                qq[j] += q[i]*f
        else:
            for j in range(i-1,i+2):
                pp[j] += (p[i]+(0<j<4)*q[i])*e
                qq[j] += q[i]*e
    p = [v%z for v in pp]
    q = [v%z for v in qq]
print(sum(p)%z)
0