結果
問題 | No.748 yuki国のお財布事情 |
ユーザー | maspy |
提出日時 | 2020-03-04 18:59:36 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 339 ms / 2,000 ms |
コード長 | 1,428 bytes |
コンパイル時間 | 119 ms |
コンパイル使用メモリ | 13,056 KB |
実行使用メモリ | 49,900 KB |
最終ジャッジ日時 | 2024-10-14 00:23:54 |
合計ジャッジ時間 | 6,025 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 26 |
ソースコード
#!/usr/bin/env python3# %%import sysread = sys.stdin.buffer.readreadline = sys.stdin.buffer.readlinereadlines = sys.stdin.buffer.readlinesfrom operator import itemgetter# %%N, M, K = map(int, readline().split())data = list(map(int, read().split()))A = data[: 3 * M:3]B = data[1: 3 * M:3]C = data[2: 3 * M:3]E = data[3 * M:]E = [e - 1 for e in E]# %%class UnionFind:def __init__(self, N):self.root = list(range(N))self.size = [1] * (N)def find_root(self, x):root = self.rootwhile root[x] != x:root[x] = root[root[x]]x = root[x]return xdef merge(self, x, y):x = self.find_root(x)y = self.find_root(y)if x == y:return Falsesx, sy = self.size[x], self.size[y]if sx < sy:self.root[x] = yself.size[y] += sxelse:self.root[y] = xself.size[x] += syreturn True# %%uf = UnionFind(N + 1)merge = uf.mergefor i in E:a, b, c = A[i], B[i], C[i]merge(a, b)seE = set(E)restA = [x for i, x in enumerate(A) if i not in seE]restB = [x for i, x in enumerate(B) if i not in seE]restC = [x for i, x in enumerate(C) if i not in seE]# %%answer = 0restABC = sorted(zip(restA, restB, restC), key=itemgetter(2))for a, b, c in restABC:if not merge(a, b):answer += cprint(answer)