結果
問題 | No.748 yuki国のお財布事情 |
ユーザー | mkawa2 |
提出日時 | 2021-02-05 01:44:26 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 463 ms / 2,000 ms |
コード長 | 1,766 bytes |
コンパイル時間 | 84 ms |
コンパイル使用メモリ | 10,788 KB |
実行使用メモリ | 52,732 KB |
最終ジャッジ日時 | 2023-09-13 22:07:24 |
合計ジャッジ時間 | 6,302 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 16 ms
8,224 KB |
testcase_01 | AC | 15 ms
8,176 KB |
testcase_02 | AC | 17 ms
8,052 KB |
testcase_03 | AC | 16 ms
8,096 KB |
testcase_04 | AC | 16 ms
8,188 KB |
testcase_05 | AC | 15 ms
8,176 KB |
testcase_06 | AC | 15 ms
8,108 KB |
testcase_07 | AC | 15 ms
8,072 KB |
testcase_08 | AC | 15 ms
8,056 KB |
testcase_09 | AC | 16 ms
8,184 KB |
testcase_10 | AC | 20 ms
8,176 KB |
testcase_11 | AC | 16 ms
8,116 KB |
testcase_12 | AC | 16 ms
8,212 KB |
testcase_13 | AC | 61 ms
14,256 KB |
testcase_14 | AC | 93 ms
18,064 KB |
testcase_15 | AC | 71 ms
15,492 KB |
testcase_16 | AC | 172 ms
26,160 KB |
testcase_17 | AC | 369 ms
49,036 KB |
testcase_18 | AC | 436 ms
51,644 KB |
testcase_19 | AC | 463 ms
52,732 KB |
testcase_20 | AC | 412 ms
47,540 KB |
testcase_21 | AC | 434 ms
50,844 KB |
testcase_22 | AC | 16 ms
8,120 KB |
testcase_23 | AC | 16 ms
8,264 KB |
testcase_24 | AC | 16 ms
8,100 KB |
testcase_25 | AC | 363 ms
43,920 KB |
testcase_26 | AC | 433 ms
52,404 KB |
testcase_27 | AC | 432 ms
52,008 KB |
testcase_28 | AC | 301 ms
37,516 KB |
ソースコード
import sys sys.setrecursionlimit(10**6) int1 = lambda x: int(x)-1 p2D = lambda x: print(*x, sep="\n") def II(): return int(sys.stdin.buffer.readline()) def MI(): return map(int, sys.stdin.buffer.readline().split()) def MI1(): return map(int1, sys.stdin.buffer.readline().split()) def LI(): return list(map(int, sys.stdin.buffer.readline().split())) def LI1(): return list(map(int1, sys.stdin.buffer.readline().split())) def LLI(rows_number): return [LI() for _ in range(rows_number)] def BI(): return sys.stdin.buffer.readline().rstrip() def SI(): return sys.stdin.buffer.readline().rstrip().decode() # dij = [(0, 1), (-1, 0), (0, -1), (1, 0)] dij = [[(-1, 0), (1, 0)], [(0, -1), (0, 1)]] inf = 10**16 # md = 998244353 md = 10**9+7 class UnionFind: def __init__(self, n): self.state = [-1] * n # self.size_table = [1] * n # cntはグループ数 # self.cnt = n def root(self, u): v = self.state[u] if v < 0: return u self.state[u] = res = self.root(v) return res def merge(self, u, v): ru = self.root(u) rv = self.root(v) if ru == rv: return du = self.state[ru] dv = self.state[rv] if du > dv: ru, rv = rv, ru if du == dv: self.state[ru] -= 1 self.state[rv] = ru # self.cnt -= 1 # self.size_table[ru] += self.size_table[rv] return n,m,k=MI() abc=LLI(m) uf=UnionFind(n) keep=[False]*m for _ in range(k): i=II()-1 keep[i]=True a,b,c=abc[i] uf.merge(a-1,b-1) cabi=[(c,a-1,b-1,i) for i,(a,b,c) in enumerate(abc)] cabi.sort(key=lambda x:x[0]) ans=0 for c,a,b,i in cabi: if keep[i]:continue if uf.root(a)==uf.root(b): ans+=c else: uf.merge(a,b) print(ans)