結果
問題 |
No.444 旨味の相乗効果
|
ユーザー |
![]() |
提出日時 | 2021-03-29 17:16:06 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 729 bytes |
コンパイル時間 | 185 ms |
コンパイル使用メモリ | 82,728 KB |
実行使用メモリ | 337,692 KB |
最終ジャッジ日時 | 2024-11-29 10:08:32 |
合計ジャッジ時間 | 58,301 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 TLE * 1 |
other | AC * 3 WA * 3 RE * 3 TLE * 14 |
ソースコード
def matrix_power(A,p): temp=A bench=p-1 ret=A while bench>0: if bench%2: ret=matrix_product(ret,temp) bench=bench//2 temp=matrix_product(temp,temp) return ret def matrix_product(A,B): ret=[[0 for i in range(len(B[0]))] for i in range(len(A))] for i in range(len(A)): for j in range(len(B[0])): s=0 for k in range(len(A[i])): s+=A[i][k]*B[k][j] ret[i][j]=s return ret n,c=map(int,input().split()) a=list(map(int,input().split())) A=[[0 for i in range(n)] for i in range(n)] b=[[1] for i in range(n)] for i in range(n): for j in range(i+1): A[i][j]=a[j] ans=matrix_product(matrix_power(A,c),b)[n-1][0] for i in range(n): ans-=a[i]**c print(ans)