結果

問題 No.1690 Power Grid
ユーザー tamato
提出日時 2021-09-24 21:56:25
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 714 ms / 3,000 ms
コード長 1,387 bytes
コンパイル時間 159 ms
コンパイル使用メモリ 82,312 KB
実行使用メモリ 107,220 KB
最終ジャッジ日時 2024-07-05 10:31:39
合計ジャッジ時間 8,993 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff #

mod = 1000000007
eps = 10**-9
inf = 10 ** 17


def main():
    import sys
    input = sys.stdin.readline

    N, M, K = map(int, input().split())
    A = list(map(int, input().split()))
    dist = [[inf] * N for _ in range(N)]
    for _ in range(M):
        x, y, z = map(int, input().split())
        x -= 1
        y -= 1
        dist[x][y] = z
        dist[y][x] = z
    for v in range(N):
        dist[v][v] = 0

    for k in range(N):
        for i in range(N):
            for j in range(N):
                dist[i][j] = min(dist[i][j], dist[i][k] + dist[k][j])

    dp = [inf] * (1 << N)
    dp[0] = 0
    for _ in range(K):
        dp_new = [inf] * (1 << N)
        for state in range(1 << N):
            if dp[state] >= inf:
                continue
            for i in range(N):
                if state >> i & 1:
                    continue
                state_new = state | (1 << i)
                if state == 0:
                    dp_new[state_new] = min(dp_new[state_new], dp[state] + A[i])
                else:
                    min_cost = inf
                    for j in range(N):
                        if state >> j & 1:
                            min_cost = min(min_cost, dist[i][j])
                    dp_new[state_new] = min(dp_new[state_new], dp[state] + A[i] + min_cost)
        dp = dp_new
    print(min(dp))


if __name__ == '__main__':
    main()
0