結果

問題 No.748 yuki国のお財布事情
ユーザー atudn
提出日時 2022-11-27 22:01:30
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 116 ms / 2,000 ms
コード長 2,836 bytes
コンパイル時間 1,811 ms
コンパイル使用メモリ 207,444 KB
最終ジャッジ日時 2025-02-09 01:39:32
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define SELECTER(_1,_2,_3,SELECT,...) SELECT
#define rep1(i,n) for(int i=0;i<(int)n;++i)
#define rep2(i,a,n) for(int i=(int)a;i<(int)n;++i)
#define rep(...) SELECTER(__VA_ARGS__,rep2,rep1) (__VA_ARGS__)
#define fi first
#define se second
using namespace std;
using ll=long long;
using pii=pair<int,int>;
using pll=pair<long long,long long>;
using veci=vector<int>;
using vecpii=vector<pair<int,int>>;
using vecll=vector<long long>;
using vecpll=vector<pair<long long,long long>>;
using vecs=vector<string>;
using vecb=vector<bool>;
using vecd=vector<double>;
template<class T> inline bool chmax(T& a,T b){if(a<b) {a=b;return true;} return false;}
template<class T> inline bool chmin(T& a,T b){if(a>b) {a=b;return true;} return false;}
 
constexpr double PI=3.14159265359;
constexpr int iMAX=2147483647;
constexpr ll lMAX=9223372036854775807;
//constexpr int INF=1<<30;
constexpr ll INF=1LL<<60;
//constexpr int MOD=1000000007;
//constexpr ll MOD=1000000007;
constexpr ll MOD=998244353;
//constexpr int LOG=62;

struct UnionFind{
	veci par_size;
	int _n;

	UnionFind(int n):_n(n),par_size(n,-1){}

	int root(int a){
		assert(0 <= a && a < _n);
		if(par_size[a]<0) return a;
		return par_size[a]=root(par_size[a]);
	}

	void unite(int a,int b){
		assert(0 <= a && a < _n);
    assert(0 <= b && b < _n);
		int ra=root(a);int rb=root(b);
		if(ra==rb) return;
		if(-par_size[ra]<-par_size[rb]) swap(ra,rb);
		par_size[ra]+=par_size[rb];
		par_size[rb]=ra;
	}

	bool same(int a,int b){
		assert(0 <= a && a < _n);
    assert(0 <= b && b < _n);
		return root(a)==root(b);
	}

	int size(int a) {
    assert(0 <= a && a < _n);
    return -par_size[root(a)];
  }

	vector<veci> groups() {
    veci leader_buf(_n), group_size(_n);
    for (int i = 0; i < _n; i++) {
    	leader_buf[i] = root(i);
      group_size[leader_buf[i]]++;
    }
    vector<veci> result(_n);
    for (int i = 0; i < _n; i++) {
      result[i].reserve(group_size[i]);
    }
    for (int i = 0; i < _n; i++) {
      result[leader_buf[i]].push_back(i);
    }
    result.erase(
      remove_if(result.begin(), result.end(),
              [&](const veci& v) {return v.empty();}),
      result.end());
    return result;
  }
};

int main(){
	//ifstream in("input.txt");
	//cin.rdbuf(in.rdbuf());
	int n,m,k;cin>>n>>m>>k;
	veci a(m),b(m),c(m);
	ll sum=0;
	rep(i,m) {
		cin>>a[i]>>b[i]>>c[i];
		a[i]--;b[i]--;
	}
	UnionFind uni(n);
	vecb used(m,false);
	rep(i,k){
		int v;cin>>v;
		v--;
		uni.unite(a[v],b[v]);
		used[v]=true;
	}

	priority_queue<pair<int,pii>,vector<pair<int,pii>>,greater<pair<int,pii>>> que;
	rep(i,m){
		if(!used[i]) que.emplace(c[i],pii{a[i],b[i]});
	}
	ll ans=0;
	while(que.size()){
		int cv=que.top().fi;
		pii v=que.top().se;
		que.pop();
		if(uni.same(v.fi,v.se)) {ans+=cv;continue;}
		uni.unite(v.fi,v.se);
	}
	cout<<ans<<endl;
}
0