結果

問題 No.1879 How many matchings?
ユーザー lam6er
提出日時 2025-04-16 16:48:40
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 809 bytes
コンパイル時間 479 ms
コンパイル使用メモリ 81,748 KB
実行使用メモリ 53,268 KB
最終ジャッジ日時 2025-04-16 16:50:51
合計ジャッジ時間 1,365 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 7 WA * 8
権限があれば一括ダウンロードができます

ソースコード

diff #

MOD = 10**9 + 7

def matrix_mult(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(matrix, power):
    result = [[1, 0], [0, 1]]  # Identity matrix
    while power > 0:
        if power % 2 == 1:
            result = matrix_mult(result, matrix)
        matrix = matrix_mult(matrix, matrix)
        power //= 2
    return result

def fib(n):
    if n == 0:
        return 0
    m = [[1, 1], [1, 0]]
    m_pow = matrix_pow(m, n-1)
    return m_pow[0][0]

n = int(input())
if n == 1:
    print(1)
elif n % 2 == 0:
    k = n // 2
    print(fib(k + 1) % MOD)
else:
    k = (n + 1) // 2
    print(fib(k + 1) % MOD)
0