結果
| 問題 |
No.1690 Power Grid
|
| コンテスト | |
| ユーザー |
rlangevin
|
| 提出日時 | 2023-02-14 23:37:54 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 667 ms / 3,000 ms |
| コード長 | 932 bytes |
| コンパイル時間 | 159 ms |
| コンパイル使用メモリ | 82,164 KB |
| 実行使用メモリ | 78,620 KB |
| 最終ジャッジ日時 | 2024-07-17 11:15:53 |
| 合計ジャッジ時間 | 12,163 ms |
|
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 3 |
| other | AC * 25 |
ソースコード
def popcount(n):
cnt = 0
while n:
cnt += n & 1
n //= 2
return cnt
N, M, K = map(int, input().split())
A = list(map(int, input().split()))
inf = 10 ** 18
G = [[inf] * N for i in range(N)]
for i in range(M):
X, Y, Z = map(int, input().split())
X, Y = X - 1, Y - 1
G[X][Y] = Z
G[Y][X] = Z
for k in range(N):
for i in range(N):
for j in range(N):
G[i][j] = min(G[i][j], G[i][k] + G[k][j])
N2 = 1 << N
dp = [inf] * N2
for i in range(N):
dp[1 << i] = A[i]
for s in range(1, N2):
for i in range(N):
if (s >> i) & 1:
continue
d = inf
for j in range(N):
if (s >> j) & 1:
d = min(d, G[i][j])
dp[s | 1 << i] = min(dp[s | 1 << i], dp[s] + A[i] + d)
ans = inf
for s in range(1, N2):
if popcount(s) == K:
ans = min(ans, dp[s])
print(ans)
rlangevin