結果
問題 | No.1595 The Final Digit |
ユーザー |
|
提出日時 | 2021-07-09 21:45:42 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 51 ms / 2,000 ms |
コード長 | 1,019 bytes |
コンパイル時間 | 185 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 59,776 KB |
最終ジャッジ日時 | 2024-07-01 15:46:56 |
合計ジャッジ時間 | 2,093 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 17 |
ソースコード
import sys input = lambda :sys.stdin.readline()[:-1] ni = lambda :int(input()) na = lambda :list(map(int,input().split())) sys.setrecursionlimit(10**7) yes = lambda :print("yes");Yes = lambda :print("Yes") no = lambda :print("no");No = lambda :print("No") ####################################################################### p,q,r,K = na() mod = 10 def mat_mul(a,b): n,m,l=len(a),len(b[0]),len(b) c=[[0]*m for _ in range(n)] for i in range(n): for k in range(l): for j in range(m): c[i][j]=(c[i][j]+a[i][k]*b[k][j])%mod return c #a^nの計算 def mat_pow(a,n): m=len(a) b=[[0]*m for _ in range(m)] for i in range(m): b[i][i]=1 while n>0: if n&1: b=mat_mul(b,a) a=mat_mul(a,a) n>>=1 return b #print(z) a = [p%10,q%10,r%10] if K>=4: z = [[1,1,1],[1,0,0],[0,1,0]] ans=mat_pow(z,K-3) print(sum([ans[0][i]*a[2-i]for i in range(3)])%10) #print(ans) #print(b) else: print(a[K-1])