結果
| 問題 | No.444 旨味の相乗効果 |
| ユーザー |
 ytft
|
| 提出日時 | 2021-03-29 17:16:06 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
TLE
|
| 実行時間 | - |
| コード長 | 729 bytes |
| コンパイル時間 | 170 ms |
| コンパイル使用メモリ | 82,304 KB |
| 実行使用メモリ | 106,624 KB |
| 最終ジャッジ日時 | 2024-05-06 23:17:49 |
| 合計ジャッジ時間 | 4,564 ms |
|
ジャッジサーバーID （参考情報） |
judge3 / judge2 |
（要ログイン）
テストケース
テストケース表示| 入力 | 結果 | 実行時間 実行使用メモリ |
|---|---|---|
| testcase_00 | AC | 41 ms
51,968 KB |
| testcase_01 | AC | 41 ms
51,840 KB |
| testcase_02 | AC | 45 ms
52,608 KB |
| testcase_03 | AC | 41 ms
52,224 KB |
| testcase_04 | TLE | - |
| testcase_05 | -- | - |
| testcase_06 | -- | - |
| testcase_07 | -- | - |
| testcase_08 | -- | - |
| testcase_09 | -- | - |
| testcase_10 | -- | - |
| testcase_11 | -- | - |
| testcase_12 | -- | - |
| testcase_13 | -- | - |
| testcase_14 | -- | - |
| testcase_15 | -- | - |
| testcase_16 | -- | - |
| testcase_17 | -- | - |
| testcase_18 | -- | - |
| testcase_19 | -- | - |
| testcase_20 | -- | - |
| testcase_21 | -- | - |
| testcase_22 | -- | - |
| testcase_23 | -- | - |
| testcase_24 | -- | - |
| testcase_25 | -- | - |
| testcase_26 | -- | - |
| testcase_27 | -- | - |
ソースコード
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)