結果
問題 | No.891 隣接3項間の漸化式 |
ユーザー |
![]() |
提出日時 | 2023-03-12 19:19:17 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,931 bytes |
コンパイル時間 | 234 ms |
コンパイル使用メモリ | 82,052 KB |
実行使用メモリ | 89,492 KB |
最終ジャッジ日時 | 2024-09-18 07:11:09 |
合計ジャッジ時間 | 6,457 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 35 WA * 4 |
ソースコード
def main(): try: import pypyjit pypyjit.set_param('max_unroll_recursion=-1') except: pass try: import sys sys.setrecursionlimit(10**7) except: pass from math import sqrt,sin,cos,tan,ceil,radians,floor,gcd,exp,log,log10,log2,factorial,fsum from heapq import heapify, heappop, heappush from bisect import bisect_left, bisect_right from copy import deepcopy import copy import random from collections import deque,Counter,defaultdict from itertools import permutations,combinations from decimal import Decimal,ROUND_HALF_UP #tmp = Decimal(mid).quantize(Decimal('0'), rounding=ROUND_HALF_UP) from functools import lru_cache, reduce #@lru_cache(maxsize=None) from operator import add,sub,mul,xor,and_,or_,itemgetter INF = 10**18 mod1 = 10**9+7 mod2 = 998244353 #DecimalならPython ''' 自明解は避ける ''' def matmul(a, b, mod): m = len(a) l = len(a[0]) n = len(b[0]) ret = [[0] * n for _ in range(m)] for i in range(m): for j in range(n): e = 0 for k in range(l): e = (e + a[i][k] * b[k][j]) % mod ret[i][j] = e return ret def matpow(X, exp, mod): Y = [[int(i == j) for j in range(len(X))] for i in range(len(X))] while exp > 0: if exp & 1: Y = matmul(X, Y, mod) X = matmul(X, X, mod) exp >>= 1 return Y a,b,n = map(int, input().split()) A = [[a,b],[1,0]] B = matpow(A,n-1,mod1) print(B[0][0]%mod1) if __name__ == '__main__': main()