結果

問題 No.2874 Gunegune Tree
ユーザー ニックネーム
提出日時 2024-09-06 23:17:03
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 190 ms / 2,000 ms
コード長 619 bytes
コンパイル時間 287 ms
コンパイル使用メモリ 82,248 KB
実行使用メモリ 76,856 KB
最終ジャッジ日時 2024-09-06 23:17:30
合計ジャッジ時間 4,920 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

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