結果

問題 No.748 yuki国のお財布事情
ユーザー prd_xxx
提出日時 2018-10-19 22:43:52
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 891 ms / 2,000 ms
コード長 1,336 bytes
コンパイル時間 99 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 43,392 KB
最終ジャッジ日時 2024-11-18 21:50:35
合計ジャッジ時間 9,333 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff #

N,M,K = map(int,input().split())
src = [tuple(map(int,input().split())) for i in range(M)]
required = [0]*M
for i in range(K):
    e = int(input()) - 1
    required[e] = 1

class UnionFind:
    def __init__(self,N):
        self.parent = [i for i in range(N)]
        self.rank = [0] * N
        self.count = 0
    def root(self,a):
        if self.parent[a] == a:
            return a
        else:
            self.parent[a] = self.root(self.parent[a])
            return self.parent[a]
    def is_same(self,a,b):
        return self.root(a) == self.root(b)
    def unite(self,a,b):
        ra = self.root(a)
        rb = self.root(b)
        if ra == rb: return
        if self.rank[ra] < self.rank[rb]:
            self.parent[ra] = rb
        else:
            self.parent[rb] = ra
            if self.rank[ra] == self.rank[rb]: self.rank[ra] += 1
        self.count += 1

uf = UnionFind(N)
for i in range(M):
    if not required[i]: continue
    a,b,c = src[i]
    uf.unite(a-1, b-1)

total_cost = 0
es = []
for i in range(M):
    if required[i]: continue
    a,b,c = src[i]
    total_cost += c
    es.append((c,i))
es.sort()

need_cost = 0
ei = 0
while uf.count < N-1:
    c,i = es[ei]
    a,b,c = src[i]
    if not uf.is_same(a-1, b-1):
        uf.unite(a-1, b-1)
        need_cost += c
    ei += 1
print(total_cost - need_cost)
0