結果

問題 No.1690 Power Grid
ユーザー titia
提出日時 2021-09-24 23:44:00
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 767 ms / 3,000 ms
コード長 1,402 bytes
コンパイル時間 304 ms
コンパイル使用メモリ 82,600 KB
実行使用メモリ 78,848 KB
最終ジャッジ日時 2024-07-05 11:37:53
合計ジャッジ時間 13,315 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline

N,M,K=map(int,input().split())
A=list(map(int,input().split()))


Distance=[[float("inf") for i in range(N)] for j in range(N)]

for i in range(N):
    Distance[i][i]=0

for i in range(M):
    x,y,z=map(int,input().split())
    x-=1
    y-=1

    Distance[x][y]=min(Distance[x][y],z)
    Distance[y][x]=min(Distance[y][x],z)

for k in range(N): # k個までの町を使ってのDisが知れているときに
    for i in range(N): # 町iと
        for j in range(i,N): # 町jとの最短距離は、
            length=Distance[i][k]+Distance[j][k]
            if Distance[i][j]>length:
                Distance[i][j]=Distance[j][i]=length

DP=[1<<60]*(1<<N)
DP[0]=0

for i in range(1,1<<N):
    c=0
    for j in range(N):
        if i & (1<<j) != 0:
            c+=1

    if c==1:
        for j in range(N):
            if i & (1<<j) !=0:
                DP[i]=A[j]
                break
    else:
        for j in range(N):
            if i & (1<<j) == 0:
                continue

            MINDIS=1<<60

            for k in range(N):
                if j!=k and (i & (1<<k) != 0):
                    MINDIS=min(MINDIS,Distance[j][k])

            DP[i]=min(DP[i],DP[i^(1<<j)]+A[j]+MINDIS)

ANS=1<<60
for i in range(1,1<<N):
    c=0
    for j in range(N):
        if i & (1<<j) != 0:
            c+=1

    if c==K:
        ANS=min(ANS,DP[i])

print(ANS)
0