結果
問題 | No.1275 綺麗な式 |
ユーザー |
![]() |
提出日時 | 2021-07-11 17:44:55 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 49 ms / 2,000 ms |
コード長 | 1,155 bytes |
コンパイル時間 | 188 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 54,912 KB |
最終ジャッジ日時 | 2024-07-02 03:09:38 |
合計ジャッジ時間 | 4,600 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 60 |
ソースコード
import sysimport mathfrom collections import deque,Counter#sys.setrecursionlimit(10**7)int1=lambda x: int(x)-1inp=lambda :int(input())mi=lambda :map(int,input().split())li=lambda :list(mi())mi1=lambda :map(int1,input().split())li1=lambda :list(mi1())mis=lambda :map(str,input().split())lis=lambda :list(mis())from collections import defaultdict"""d=defaultdict(int) #初期値 0d=defaultdict(lambda:1) #初期値 1"""mod=10**9+7Mod=998244353INF=10**18ans=0def matmul(A,B,mod): # A,B: 行列res = [[0]*len(B[0]) for _ in [None]*len(A)]for i, resi in enumerate(res):for k, aik in enumerate(A[i]):for j,bkj in enumerate(B[k]):resi[j] += aik*bkjresi[j] %= modreturn resdef matpow(A,p,mod): #A^p mod Mif p%2:return matmul(A, matpow(A,p-1,mod),mod)elif p > 0:b = matpow(A,p//2,mod)return matmul(b,b,mod)else:return [[int(i==j) for j in range(len(A))] for i in range(len(A))]a,b=mi()n=inp()if n==0:print(2)exit()x=[[2*a,b-a**2],[1,0]]y=matpow(x,n-1,mod)z=matmul(y,[[2*a],[2]],mod)print(z[0][0]%mod)