結果
問題 | No.748 yuki国のお財布事情 |
ユーザー |
|
提出日時 | 2022-06-12 19:56:59 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 547 ms / 2,000 ms |
コード長 | 755 bytes |
コンパイル時間 | 206 ms |
コンパイル使用メモリ | 81,992 KB |
実行使用メモリ | 104,824 KB |
最終ジャッジ日時 | 2024-09-23 06:28:10 |
合計ジャッジ時間 | 8,114 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 26 |
ソースコード
import syssys.setrecursionlimit(10 ** 8)N,M,K = map(int,input().split())edge = []for _ in range(M):a,b,c = map(int,input().split())edge.append((c,a,b,_))parent = list(range(N + 1))def find(i):if parent[i] == i:return iparent[i] = find(parent[i])return parent[i]def unite(i,j):I = find(i)J = find(j)if I == J:return Falseparent[I] = Jparent[i] = Jreturn Truedef isSame(i,j):return find(i) == find(j)ans = 0s = set()for _ in range(K):e = int(input())c,a,b,i = edge[e - 1]unite(a,b)s.add(e - 1)edge.sort(key = lambda x:x[0])for c,a,b,_ in edge:if _ in s:continueif isSame(a,b):ans += celse:unite(a,b)print(ans)