結果

問題 No.1810 RGB Biscuits
コンテスト
ユーザー rin204
提出日時 2022-01-21 01:10:05
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 238 ms / 2,000 ms
コード長 762 bytes
コンパイル時間 271 ms
コンパイル使用メモリ 82,376 KB
実行使用メモリ 84,716 KB
最終ジャッジ日時 2024-11-24 12:31:18
合計ジャッジ時間 4,100 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

from copy import deepcopy

MOD = 10 ** 9 + 7

a, b = map(int, input().split())
n = int(input())
for _ in range(n):
    t = int(input())
    A = [[a, b], [1, 0]]
    B = [1, 1]
    x = t // 2
    while x:
        if x & 1:
            C = [0, 0]
            for i in range(2):
                for j in range(2):
                    C[i] += A[i][j] * B[j]
                C[i] %= MOD
            B = C.copy()
        x >>= 1
        C = [[0, 0], [0, 0]]
        for i in range(2):
            for j in range(2):
                for k in range(2):
                    C[i][j] += A[i][k] * A[k][j]
                C[i][j] %= MOD
        A = deepcopy(C)
    
    if t % 2 == 0:
        print(sum(B) % MOD)
    else:
        print((sum(B) + B[0] * a + B[1] * b) % MOD)
0