結果

問題 No.3228 Very Large Fibonacci Sum
ユーザー tassei903
提出日時 2025-08-08 21:49:33
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,074 bytes
コンパイル時間 403 ms
コンパイル使用メモリ 82,904 KB
実行使用メモリ 62,672 KB
最終ジャッジ日時 2025-08-08 21:49:43
合計ジャッジ時間 2,188 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 21 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
input = lambda :sys.stdin.readline()[:-1]
ni = lambda :int(input())
na = lambda :list(map(int,input().split()))
yes = lambda :print("yes");Yes = lambda :print("Yes");YES = lambda : print("YES")
no = lambda :print("no");No = lambda :print("No");NO = lambda : print("NO")
#######################################################################
mod = 10 ** 9 + 7
def mat_prod(A,B):
    C = [[0]*(len(B[0])) for _ in range(len(A))]
    for i,Ci in enumerate(C):
        for k,aik in enumerate(A[i]):
            for j,bkj in enumerate(B[k]):
                Ci[j] = (Ci[j]+aik*bkj)%mod
    return C

def mat_pow(A,p):
  if p%2:
    return mat_prod(A,mat_pow(A,p-1))
  elif p > 0:
    b = mat_pow(A,p//2)
    return mat_prod(b,b)
  else:
    return [[int(i == j) for j in range(len(A))] for i in range(len(A))]
  

a, b, c, d, e, n = na()
if n == 0:
   print(a)
elif n == 1:
   print(b)
else:
    A = [[1, c, d, e], [0, c, d, e], [0, 1, 0, 0], [0, 0, 0, 1]]

    F = mat_pow(A, n-1)
    x = ([a + b], [b], [a], [1])
    
    y = mat_prod(F, x)
    print(y[0][0] % mod)
0