結果
問題 |
No.748 yuki国のお財布事情
|
ユーザー |
|
提出日時 | 2023-01-29 02:35:39 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 516 ms / 2,000 ms |
コード長 | 1,078 bytes |
コンパイル時間 | 142 ms |
コンパイル使用メモリ | 82,436 KB |
実行使用メモリ | 115,980 KB |
最終ジャッジ日時 | 2024-06-29 02:38:24 |
合計ジャッジ時間 | 7,597 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 26 |
ソースコード
import heapq class UnionFind: def __init__(self, size): self.parent = [-1] * size def union(self, x, y): x, y = self.root(x), self.root(y) if x == y: return False if self.parent[x] > self.parent[y]: x, y = y, x self.parent[x] += self.parent[y] self.parent[y] = x return True def root(self, x): if self.parent[x] < 0: return x self.parent[x] = self.root(self.parent[x]) return self.parent[x] N, M, K = map(int, input().split()) graph = [[] for _ in range(N)] edges = [] uf = UnionFind(N) for i in range(M): u, v, w = map(int, input().split()) u -= 1 v -= 1 graph[u].append((v, w)) graph[v].append((u, w)) edges.append((u, v, w, i)) used = set() for _ in range(K): e = int(input()) - 1 u, v, w, _ = edges[e] uf.union(u, v) used.add(e) edges.sort(key=lambda x: x[2]) erased = 0 for u, v, w, i in edges: if i in used: continue if not uf.union(u, v): erased += w print(erased)