結果
問題 | No.748 yuki国のお財布事情 |
ユーザー | qib |
提出日時 | 2022-11-13 12:37:19 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 501 ms / 2,000 ms |
コード長 | 1,195 bytes |
コンパイル時間 | 285 ms |
コンパイル使用メモリ | 86,976 KB |
実行使用メモリ | 97,116 KB |
最終ジャッジ日時 | 2023-10-12 21:55:25 |
合計ジャッジ時間 | 9,337 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge11 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 74 ms
71,084 KB |
testcase_01 | AC | 74 ms
71,088 KB |
testcase_02 | AC | 74 ms
71,172 KB |
testcase_03 | AC | 73 ms
71,344 KB |
testcase_04 | AC | 72 ms
71,340 KB |
testcase_05 | AC | 72 ms
71,000 KB |
testcase_06 | AC | 73 ms
70,968 KB |
testcase_07 | AC | 72 ms
71,124 KB |
testcase_08 | AC | 74 ms
70,972 KB |
testcase_09 | AC | 74 ms
71,048 KB |
testcase_10 | AC | 73 ms
70,800 KB |
testcase_11 | AC | 74 ms
71,080 KB |
testcase_12 | AC | 74 ms
71,092 KB |
testcase_13 | AC | 170 ms
80,436 KB |
testcase_14 | AC | 213 ms
83,068 KB |
testcase_15 | AC | 175 ms
80,628 KB |
testcase_16 | AC | 268 ms
85,432 KB |
testcase_17 | AC | 454 ms
95,020 KB |
testcase_18 | AC | 466 ms
96,112 KB |
testcase_19 | AC | 449 ms
97,116 KB |
testcase_20 | AC | 404 ms
94,332 KB |
testcase_21 | AC | 501 ms
96,156 KB |
testcase_22 | AC | 72 ms
71,024 KB |
testcase_23 | AC | 72 ms
71,172 KB |
testcase_24 | AC | 72 ms
71,152 KB |
testcase_25 | AC | 374 ms
94,072 KB |
testcase_26 | AC | 453 ms
96,020 KB |
testcase_27 | AC | 482 ms
96,712 KB |
testcase_28 | AC | 299 ms
89,920 KB |
ソースコード
class UnionFind: def __init__(self, n): self.node = [-1 for _ in range(n)] def root(self, v): if self.node[v] < 0: return v st = [] while self.node[v] >= 0: st.append(v) v = self.node[v] for u in st: self.node[u] = v return v def size(self, v): v = self.root(v) return (- self.node[v]) def same(self, u, v): return self.root(u) == self.root(v) def unite(self, u, v): ru = self.root(u) rv = self.root(v) if ru == rv: return du = self.node[ru] dv = self.node[rv] if du <= dv: self.node[rv] = ru self.node[ru] += dv else: self.node[ru] = rv self.node[rv] += du n, m, k = map(int, input().split()) edges = [] for i in range(m): a, b, c = map(int, input().split()) a -= 1 b -= 1 edges.append((a, b, c, i)) used = [False for _ in range(m)] e = [int(input()) - 1 for _ in range(k)] uf = UnionFind(n) for ei in e: u, v, _, _ = edges[ei] uf.unite(u, v) used[ei] = True edges.sort(key=lambda x: x[2]) ans = 0 for i in range(m): a, b, c, ei = edges[i] if used[ei]: continue if uf.same(a, b): ans += c else: uf.unite(a, b) print(ans)