結果

問題 No.572 妖精の演奏
ユーザー ckawatak
提出日時 2019-03-09 22:02:24
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 102 ms / 2,000 ms
コード長 927 bytes
コンパイル時間 167 ms
コンパイル使用メモリ 82,192 KB
実行使用メモリ 77,016 KB
最終ジャッジ日時 2024-06-23 15:15:27
合計ジャッジ時間 2,394 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #

N = int(input())
M = int(input())

X = [list(map(int, input().split(' '))) for _ in range(M)]
A = [[[0 for _ in range(M+1)] for _ in range(M+1)] for _ in range(M+2)]

for i in range(M):
    for j in range(M):
        A[0][i][j] = X[i][j]

for i in range(M+1):
    for x in range(M):
        for y in range(M):
            for z in range(M):
                A[i+1][x][y] = max(A[i+1][x][y], A[i][x][z]+A[i][z][y])

B = [[[0 for _ in range(M+1)] for _ in range(M+1)] for _ in range(M+2)]
answer = 0                
N = N-1
for i in range(M+1):
    if N & (1<<i):
        for x in range(M):
            for y in range(M):
                for z in range(M):
                    B[i+1][x][y] = max(B[i+1][x][y], B[i][x][z]+A[i][z][y])
                    answer = max(answer, B[i+1][x][y])
    else:
        for x in range(M):
            for y in range(M):
                B[i+1][x][y] = B[i][x][y] 

print(answer)                
0