結果

問題 No.748 yuki国のお財布事情
ユーザー prd_xxxprd_xxx
提出日時 2018-10-19 22:43:52
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 910 ms / 2,000 ms
コード長 1,336 bytes
コンパイル時間 153 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 43,264 KB
最終ジャッジ日時 2024-04-29 16:32:44
合計ジャッジ時間 9,516 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 31 ms
10,752 KB
testcase_01 AC 34 ms
10,752 KB
testcase_02 AC 31 ms
10,880 KB
testcase_03 AC 32 ms
10,752 KB
testcase_04 AC 31 ms
10,752 KB
testcase_05 AC 31 ms
10,752 KB
testcase_06 AC 31 ms
10,752 KB
testcase_07 AC 31 ms
10,752 KB
testcase_08 AC 31 ms
10,880 KB
testcase_09 AC 31 ms
10,624 KB
testcase_10 AC 31 ms
10,752 KB
testcase_11 AC 30 ms
10,752 KB
testcase_12 AC 31 ms
10,752 KB
testcase_13 AC 111 ms
14,336 KB
testcase_14 AC 183 ms
15,872 KB
testcase_15 AC 131 ms
14,976 KB
testcase_16 AC 323 ms
19,712 KB
testcase_17 AC 739 ms
36,992 KB
testcase_18 AC 805 ms
35,456 KB
testcase_19 AC 801 ms
33,408 KB
testcase_20 AC 706 ms
30,592 KB
testcase_21 AC 825 ms
37,504 KB
testcase_22 AC 31 ms
10,624 KB
testcase_23 AC 31 ms
10,752 KB
testcase_24 AC 31 ms
10,752 KB
testcase_25 AC 589 ms
34,944 KB
testcase_26 AC 910 ms
42,496 KB
testcase_27 AC 905 ms
43,264 KB
testcase_28 AC 512 ms
25,088 KB
権限があれば一括ダウンロードができます

ソースコード

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