結果
問題 | No.1690 Power Grid |
ユーザー |
![]() |
提出日時 | 2021-09-24 22:07:54 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 772 ms / 3,000 ms |
コード長 | 861 bytes |
コンパイル時間 | 241 ms |
コンパイル使用メモリ | 81,792 KB |
実行使用メモリ | 78,720 KB |
最終ジャッジ日時 | 2024-07-05 10:40:01 |
合計ジャッジ時間 | 11,909 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 25 |
ソースコード
def Warshall_Floyd(edges,N): for k in range(N): for i in range(N): for j in range(N): edges[i][j]=min(edges[i][j],edges[i][k]+edges[k][j]) return edges INF=10**18 n,m,K=map(int,input().split()) a=list(map(int,input().split())) edge=[[INF]*n for i in range(n)] for i in range(m): x,y,k=map(int,input().split()) x,y=x-1,y-1 edge[x][y]=k edge[y][x]=k for i in range(n): edge[i][i]=0 dist=Warshall_Floyd(edge,n) dp=[INF]*(1<<n) dp[0]=0 ans=INF for bit in range(1<<n): if bit==0: for j in range(n): dp[1<<j]=a[j] continue if bin(bit).count('1')==K: ans=min(ans,dp[bit]) x=[] y=[] for i in range(n): if bit&(1<<i)==0: x.append(i) else: y.append(i) for i in x: cost=INF for j in y: cost=min(cost,dist[i][j]) dp[bit+2**i]=min(dp[bit+2**i],dp[bit]+cost+a[i]) print(ans)