結果

問題 No.1275 綺麗な式
ユーザー ygd.
提出日時 2020-11-25 00:30:35
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 397 bytes
コンパイル時間 314 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 44,604 KB
最終ジャッジ日時 2024-07-23 18:58:42
合計ジャッジ時間 36,598 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 12 WA * 48
権限があれば一括ダウンロードができます

ソースコード

diff # 

import numpy as np
MOD = pow(10,9)+7
a,b = map(int,input().split())
n = int(input())
A = np.array([[2*a, b - a**2], [1, 0]])
#print(A)
MAX = 70
L = [A]
for i in range(MAX):
  A = np.dot(A,A)
  A = np.mod(A, MOD)
  L.append(A)
#print(L)
X = np.array([[1, 0], [0, 1]])
for i in range(MAX):
  if (n-1>>i)&1 == 1:
    X = np.dot(X,L[i])
    X = np.mod(X,MOD)
ans = X[0,0]*2*a + X[0,1]*2
print(ans%MOD)
0