結果

問題 No.2156 ぞい文字列
ユーザー lam6er
提出日時 2025-03-20 18:47:27
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 39 ms / 2,000 ms
コード長 836 bytes
コンパイル時間 151 ms
コンパイル使用メモリ 82,420 KB
実行使用メモリ 54,092 KB
最終ジャッジ日時 2025-03-20 18:48:45
合計ジャッジ時間 1,714 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 16
権限があれば一括ダウンロードができます

ソースコード

diff #

mod = 998244353

def multiply(a, b):
    return [
        [
            (a[0][0] * b[0][0] + a[0][1] * b[1][0]) % mod,
            (a[0][0] * b[0][1] + a[0][1] * b[1][1]) % mod
        ],
        [
            (a[1][0] * b[0][0] + a[1][1] * b[1][0]) % mod,
            (a[1][0] * b[0][1] + a[1][1] * b[1][1]) % mod
        ]
    ]

def matrix_pow(mat, power):
    result = [[1, 0], [0, 1]]  # Identity matrix
    while power > 0:
        if power % 2 == 1:
            result = multiply(result, mat)
        mat = multiply(mat, mat)
        power //= 2
    return result

def fib(m):
    if m == 1 or m == 2:
        return 1
    mat = matrix_pow([[1, 1], [1, 0]], m - 2)
    return (mat[0][0] + mat[0][1]) % mod

n = int(input())
m = n + 1
if m <= 2:
    print((0) % mod)
else:
    res = fib(m)
    ans = (res - 1) % mod
    print(ans)
0