結果
問題 | No.891 隣接3項間の漸化式 |
ユーザー |
![]() |
提出日時 | 2023-03-12 19:41:07 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 156 ms / 2,000 ms |
コード長 | 1,983 bytes |
コンパイル時間 | 468 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 89,216 KB |
最終ジャッジ日時 | 2024-09-18 07:12:22 |
合計ジャッジ時間 | 7,457 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 39 |
ソースコード
def main():try:import pypyjitpypyjit.set_param('max_unroll_recursion=-1')except:passtry:import syssys.setrecursionlimit(10**7)except:passfrom math import sqrt,sin,cos,tan,ceil,radians,floor,gcd,exp,log,log10,log2,factorial,fsumfrom heapq import heapify, heappop, heappushfrom bisect import bisect_left, bisect_rightfrom copy import deepcopyimport copyimport randomfrom collections import deque,Counter,defaultdictfrom itertools import permutations,combinationsfrom 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_,itemgetterINF = 10**18mod1 = 10**9+7mod2 = 998244353#DecimalならPython'''自明解は避ける'''def matmul(A,B,mod):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):if 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,n = map(int, input().split())A = [[a,b],[1,0]]if n == 0:print(0)elif n == 1:print(1)else:B = matpow(A,n-1,mod1)print(B[0][0]%mod1)if __name__ == '__main__':main()