結果

問題 No.1595 The Final Digit
ユーザー googol_S0googol_S0
提出日時 2021-07-09 21:30:06
言語 PyPy3
(7.3.11)
結果
AC  
実行時間 77 ms / 2,000 ms
コード長 700 bytes
コンパイル時間 514 ms
コンパイル使用メモリ 87,164 KB
実行使用メモリ 76,376 KB
最終ジャッジ日時 2023-09-14 07:51:45
合計ジャッジ時間 2,856 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge11
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 73 ms
71,380 KB
testcase_01 AC 74 ms
71,344 KB
testcase_02 AC 67 ms
71,404 KB
testcase_03 AC 68 ms
71,300 KB
testcase_04 AC 73 ms
75,404 KB
testcase_05 AC 73 ms
75,484 KB
testcase_06 AC 74 ms
76,372 KB
testcase_07 AC 74 ms
76,376 KB
testcase_08 AC 69 ms
71,064 KB
testcase_09 AC 68 ms
71,076 KB
testcase_10 AC 68 ms
71,456 KB
testcase_11 AC 69 ms
71,348 KB
testcase_12 AC 69 ms
71,408 KB
testcase_13 AC 68 ms
71,160 KB
testcase_14 AC 77 ms
76,208 KB
testcase_15 AC 75 ms
76,288 KB
testcase_16 AC 75 ms
76,204 KB
testcase_17 AC 70 ms
71,212 KB
testcase_18 AC 69 ms
71,204 KB
testcase_19 AC 76 ms
76,204 KB
権限があれば一括ダウンロードができます

ソースコード

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