結果

問題 No.2156 ぞい文字列
ユーザー ntuda
提出日時 2025-03-02 13:09:23
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 39 ms / 2,000 ms
コード長 463 bytes
コンパイル時間 375 ms
コンパイル使用メモリ 82,792 KB
実行使用メモリ 54,192 KB
最終ジャッジ日時 2025-03-02 13:09:26
合計ジャッジ時間 2,405 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 16
権限があれば一括ダウンロードができます

ソースコード

diff #

MOD = 998244353
D = 2

def mt(A, B):
    C = [[0] * D for _ in range(D)]
    for i in range(D):
        for j in range(D):
            tmp = 0
            for k in range(D):
                tmp += A[i][k] * B[k][j]
                tmp %= MOD
            C[i][j] = tmp
    return C

A = [[0, 1], [1, 1]]
X = [[1, 0], [0, 1]]
N = int(input())
N -= 1
while N > 0:
    if N & 1:
        X = mt(A, X)
    A = mt(A, A)
    N >>= 1

print((X[0][1] + X[1][1] - 1) % MOD)
0