結果

問題 No.2156 ぞい文字列
コンテスト
ユーザー isee
提出日時 2022-12-09 21:38:36
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 43 ms / 2,000 ms
コード長 782 bytes
コンパイル時間 793 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 52,736 KB
最終ジャッジ日時 2024-10-14 21:04:08
合計ジャッジ時間 2,139 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = lambda: sys.stdin.readline().rstrip()

MOD = 998244353

def even(P):
    return P[0::2]

def odd(P):
    return P[1::2]

def mul(P, Q):
    N = len(P); M = len(Q)
    R = [0]*(N+M-1)
    for i in range(N):
        for j in range(M):
            R[i+j] += P[i]*Q[j]
            R[i+j] %= MOD
    return R

def negx(Q):
    return [(-n if i&1 else n) for i, n in enumerate(Q)]

def bostan_mori(N, P, Q):
    while N > 0:
        Qx = negx(Q)
        U = mul(P, Qx)
        if N&1: P = odd(U)
        else: P = even(U)
        Q = even(mul(Q, Qx))
        N >>= 1
    return P[0] * pow(Q[0], MOD-2, MOD)

def main():
    # 入力
    N = int(input())
    # 計算・出力
    print(bostan_mori(N+1, [0, 1], [1, -1, -1]) - 1)

if __name__ == "__main__":
    main()
0