結果

問題 No.1690 Power Grid
コンテスト
ユーザー ああいい
提出日時 2022-03-04 10:54:58
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 667 ms / 3,000 ms
コード長 1,093 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 287 ms
コンパイル使用メモリ 82,068 KB
実行使用メモリ 78,636 KB
最終ジャッジ日時 2024-07-18 08:11:58
合計ジャッジ時間 12,113 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

N,M,K = map(int,input().split())
A = list(map(int,input().split()))
inf = 10 ** 12
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 i in range(N):
    dist[i][i] = 0
for k in range(N):
    for j in range(N):
        for i in range(N):
            if dist[i][j] > dist[i][k] + dist[k][j]:
                dist[i][j] = dist[i][k] + dist[k][j]
dp = [inf] * (1 << N)
for i in range(N):
    mask = 1 << i
    dp[mask] = A[i]
for bit in range(1, 1 << N):
    for i in range(N):
        mask = 1 << i
        if bit & mask == 0:
            continue
        b = bit ^ mask
        _min = inf
        index = -1
        for j in range(N):
            if b & (1 << j):
                if dist[i][j] < _min:
                    _min = dist[i][j]
                    index = j
        if dp[bit] > dp[b] + _min + A[i]:
            dp[bit] = dp[b] + _min + A[i]
ans = inf
for bit in range(1, 1 << N):
    if bin(bit).count('1') == K:
        if ans > dp[bit]:
            ans = dp[bit]
print(ans)
0