結果

問題 No.748 yuki国のお財布事情
コンテスト
ユーザー polylogK
提出日時 2018-11-22 20:59:46
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 51 ms / 2,000 ms
コード長 1,045 bytes
コンパイル時間 1,833 ms
コンパイル使用メモリ 177,248 KB
実行使用メモリ 5,864 KB
最終ジャッジ日時 2024-12-24 17:55:01
合計ジャッジ時間 4,193 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
typedef long long i64;
using std::cout;
using std::endl;
using std::cin;

struct UnionFind {
	std::vector<int> data;
	
	UnionFind(int n) {
		data.assign(n, -1);
	}
	
	bool unite(int x, int y) {
		x = find(x); y = find(y);
		if(x == y) return false;
		if(data[x] > data[y]) std::swap(x, y);
		data[x] += data[y];
		data[y] = x;
		return true;
	}
	
	int find(int k) {
		if(data[k] < 0) return k;
		return data[k] = find(data[k]);
	}
};

int main() {
	int n, m, k; scanf("%d %d %d", &n, &m, &k);
	std::vector<int> a(m), b(m), c(m);
	std::vector<std::pair<int, int>> vec;
	i64 sum = 0;
	for(int i = 0; i < m; i++) {
		scanf("%d %d %d", &a[i], &b[i], &c[i]); a[i]--; b[i]--;
		
		sum += c[i];
		vec.push_back({c[i], i});
	}
	sort(begin(vec), end(vec));
	
	i64 ans = 0;
	UnionFind uf(n);
	for(int i = 0; i < k; i++) {
		int p; scanf("%d", &p); p--;
		
		uf.unite(a[p], b[p]);
		ans += c[p];
	}
	
	for(auto v : vec) {
		auto p = v.second;
		if(uf.unite(a[p], b[p])) ans += c[p];
	}
	
	printf("%lld\n", sum - ans);
	return 0;
}
0