結果
問題 | No.1275 綺麗な式 |
ユーザー |
|
提出日時 | 2020-10-30 22:04:43 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 44 ms / 2,000 ms |
コード長 | 820 bytes |
コンパイル時間 | 217 ms |
コンパイル使用メモリ | 81,920 KB |
実行使用メモリ | 53,376 KB |
最終ジャッジ日時 | 2024-07-22 00:40:34 |
合計ジャッジ時間 | 3,872 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 60 |
ソースコード
def power(x,n,mod): digit = len(str(bin(n)))-2 a = len(x) dp = [x]*digit for i in range(1,digit): dp[i] = dot(dp[i-1],dp[i-1],mod) ans = [[0]*a for _ in range(a)] for i in range(a): ans[i][i] = 1 for i in range(digit): if 1<<i & n: ans = dot(dp[i],ans,mod) return ans def dot(A,B,mod): p = len(A) q = len(A[0]) r = len(B) s = len(B[0]) if q!=r: print('掛け算できません') return ans = [[0]*s for _ in range(p)] for i in range(p): for j in range(s): for k in range(q): ans[i][j] += A[i][k]*B[k][j] ans[i][j] %= mod return ans def solve(): a, b = map(int, input().split()) N = int(input()) init = [[1],[0]] mat = [[a,b],[1,a]] mod = 10**9+7 ans = dot(power(mat,N,mod),init,mod)[0][0] return ans*2%mod print(solve())