結果

問題 No.1595 The Final Digit
ユーザー googol_S0googol_S0
提出日時 2021-07-09 21:30:06
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 45 ms / 2,000 ms
コード長 700 bytes
コンパイル時間 164 ms
コンパイル使用メモリ 82,164 KB
実行使用メモリ 60,544 KB
最終ジャッジ日時 2024-07-01 15:16:33
合計ジャッジ時間 1,778 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

P,Q,R,K=map(int,input().split())
P%=10
Q%=10
R%=10
mod=10
def m(a,b):
  r=[[0]*len(b[0]) for i in range(len(a))]
  for i in range(len(a)):
    for k in range(len(b)):
      for j in range(len(b[0])):
        r[i][j]=(r[i][j]+a[i][k]*b[k][j])%mod
  return r

def p(a,n):
  r=[[0]*len(a) for i in range(len(a))]
  b=[]
  for i in range(len(a)):
    r[i][i]=1
    b.append(a[i][:])
  l=n
  while l>0:
    if l&1:
      r=m(b,r)
    b=m(b,b)
    l>>=1
  return r

if K==1:
  print(P)
if K==2:
  print(Q)
if K==3:
  print(R)
if K<=3:
  exit()
X=[[0,0,0],[0,0,0],[0,0,0]]
X[0][0]=1
X[0][1]=1
X[0][2]=1
X[1][0]=1
X[2][1]=1
Y=[[0]*3 for i in range(3)]
Y[0][2]=R
Y[1][2]=Q
Y[2][2]=P
print(m(p(X,K-3),Y)[0][2])
0