結果

問題 No.3036 Nauclhlt型文字列
コンテスト
ユーザー gew1fw
提出日時 2025-06-12 19:00:44
言語 PyPy3
(7.3.15)
結果
RE  
実行時間 -
コード長 1,098 bytes
コンパイル時間 213 ms
コンパイル使用メモリ 82,128 KB
実行使用メモリ 67,220 KB
最終ジャッジ日時 2025-06-12 19:00:53
合計ジャッジ時間 2,056 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample RE * 3
other RE * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

def multiply(a, b, mod):
    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, mod):
    result = [[1, 0], [0, 1]]
    while power > 0:
        if power % 2 == 1:
            result = multiply(result, mat, mod)
        mat = multiply(mat, mat, mod)
        power = power // 2
    return result

def main():
    import sys
    input = sys.stdin.read().split()
    idx = 0
    T = int(input[idx])
    idx += 1
    mod = 10**9 + 7
    for _ in range(T):
        n = int(input[idx])
        idx += 1
        if n == 0:
            print(2 % mod)
        elif n == 1:
            print(1 % mod)
        else:
            exponent = n - 1
            M = [[1, 1], [1, 0]]
            mat = matrix_pow(M, exponent, mod)
            l_n = (mat[0][0] * 1 + mat[0][1] * 2) % mod
            print(l_n)

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