結果
問題 | No.1275 綺麗な式 |
ユーザー |
![]() |
提出日時 | 2021-07-11 17:42:15 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,122 bytes |
コンパイル時間 | 239 ms |
コンパイル使用メモリ | 82,560 KB |
実行使用メモリ | 55,040 KB |
最終ジャッジ日時 | 2024-07-02 03:09:33 |
合計ジャッジ時間 | 5,105 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 58 WA * 2 |
ソースコード
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()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])