結果

問題 No.1448 和差算
ユーザー Kude
提出日時 2021-03-31 14:33:21
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 43 ms / 2,000 ms
コード長 666 bytes
コンパイル時間 465 ms
コンパイル使用メモリ 81,968 KB
実行使用メモリ 55,164 KB
最終ジャッジ日時 2024-12-14 19:13:30
合計ジャッジ時間 2,882 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff #

M = [
    [1, -1],
    [1, 1]
]
MOD = 10 ** 9 + 7
def mul(a, b):
    return [[sum(a[i][k] * b[k][j] for k in range(2)) % MOD for j in range(2)] for i in range(2)]

def matpow(x):
    ret = [[1, 0], [0, 1]]
    t = [m[:] for m in M]
    while x:
        if x & 1:
            ret = mul(ret, t)
        x >>= 1
        t = mul(t, t)
    return ret

# for i in range(10):
    # print(i)
    # print(*matpow(i), sep='\n', end='\n')

A, B, C, D = map(int, input().split())
N = int(input())
if 3 <= N % 8 <= 6:
    a = A
else:
    a = B
if 2 <= N % 8 <= 5:
    b = C
else:
    b = D
X = matpow(N)
ans = ((X[0][0] + X[1][0]) * a + (X[0][1] + X[1][1]) * b) % MOD
print(ans)
0