結果
問題 |
No.1690 Power Grid
|
ユーザー |
![]() |
提出日時 | 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)