結果

問題 No.1879 How many matchings?
ユーザー gew1fw
提出日時 2025-06-12 15:54:50
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,736 bytes
コンパイル時間 255 ms
コンパイル使用メモリ 82,224 KB
実行使用メモリ 54,796 KB
最終ジャッジ日時 2025-06-12 15:54:52
合計ジャッジ時間 1,413 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 5 WA * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 10**9 + 7

def matrix_mult(a, b):
    return [[(a[0][0]*b[0][0] + a[0][1]*b[1][0] + a[0][2]*b[2][0]) % MOD,
             (a[0][0]*b[0][1] + a[0][1]*b[1][1] + a[0][2]*b[2][1]) % MOD,
             (a[0][0]*b[0][2] + a[0][1]*b[1][2] + a[0][2]*b[2][2]) % MOD],
            [(a[1][0]*b[0][0] + a[1][1]*b[1][0] + a[1][2]*b[2][0]) % MOD,
             (a[1][0]*b[0][1] + a[1][1]*b[1][1] + a[1][2]*b[2][1]) % MOD,
             (a[1][0]*b[0][2] + a[1][1]*b[1][2] + a[1][2]*b[2][2]) % MOD],
            [(a[2][0]*b[0][0] + a[2][1]*b[1][0] + a[2][2]*b[2][0]) % MOD,
             (a[2][0]*b[0][1] + a[2][1]*b[1][1] + a[2][2]*b[2][1]) % MOD,
             (a[2][0]*b[0][2] + a[2][1]*b[1][2] + a[2][2]*b[2][2]) % MOD]]

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

def solve_odd(n):
    if n == 1:
        return 1
    elif n == 3:
        return 3
    elif n == 5:
        return 7
    elif n == 7:
        return 13
    elif n == 9:
        return 31
    k = (n - 1) // 2 - 2
    trans = [[2, 1, 0], [1, 0, 1], [0, 1, 0]]
    mat = matrix_pow(trans, k)
    init = [7, 3, 1]
    res = (mat[0][0] * init[0] + mat[0][1] * init[1] + mat[0][2] * init[2]) % MOD
    return res

n = int(input())
if n % 2 == 0:
    k = n // 2
    if k <= 1:
        print(1 % MOD)
    else:
        print((2 * (k - 1)) % MOD)
else:
    if n == 1:
        print(1)
    elif n == 3:
        print(3 % MOD)
    elif n == 5:
        print(7 % MOD)
    elif n == 7:
        print(13 % MOD)
    elif n == 9:
        print(31 % MOD)
    else:
        print(solve_odd(n) % MOD)
0