結果

問題 No.1275 綺麗な式
ユーザー marroncastle
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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())
0