結果
問題 | No.1275 綺麗な式 |
ユーザー | Eki1009 |
提出日時 | 2020-10-30 21:52:03 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 40 ms / 2,000 ms |
コード長 | 610 bytes |
コンパイル時間 | 461 ms |
コンパイル使用メモリ | 82,180 KB |
実行使用メモリ | 53,120 KB |
最終ジャッジ日時 | 2024-07-22 00:17:35 |
合計ジャッジ時間 | 4,509 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 60 |
ソースコード
import sys input = sys.stdin.readline a, b = map(int, input().split()) n = int(input()) mod = 10**9+7 def prod(A, B): n = len(A) m = len(B) l = len(B[0]) C = [[0]*l for _ in range(n)] for i in range(n): for j in range(m): for k in range(l): C[i][k] += A[i][j] * B[j][k] C[i][k] %= mod return C def mpow(X, n): if n == 0: return [[1,0], [0,1]] if n%2: return prod(X, mpow(X,n-1)) Y = mpow(X, n//2) return prod(Y, Y) X = [[a,b], [1,a]] A = [[1],[0]] L = prod(mpow(X,n),A) ans = L[0][0]*2%mod print(ans)