結果
問題 | No.748 yuki国のお財布事情 |
ユーザー |
![]() |
提出日時 | 2020-07-19 23:37:13 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 111 ms / 2,000 ms |
コード長 | 2,366 bytes |
コンパイル時間 | 2,306 ms |
コンパイル使用メモリ | 210,772 KB |
最終ジャッジ日時 | 2025-01-12 01:03:43 |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 26 |
ソースコード
#include<bits/stdc++.h>using namespace std;#define ALL(x) x.begin(),x.end()#define rep(i,n) for(int i=0;i<(n);i++)#define debug(v) cout<<#v<<":";for(auto x:v){cout<<x<<' ';}cout<<endl;#define mod 1000000007using ll=long long;const int INF=1000000000;const ll LINF=1001002003004005006ll;int dx[]={1,0,-1,0};int dy[]={0,1,0,-1};// ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}template<class T>bool chmax(T &a,const T &b){if(a<b){a=b;return true;}return false;}template<class T>bool chmin(T &a,const T &b){if(b<a){a=b;return true;}return false;}struct IOSetup{IOSetup(){cin.tie(0);ios::sync_with_stdio(0);cout<<fixed<<setprecision(12);}} iosetup;template<typename T1,typename T2>ostream &operator<<(ostream &os,const pair<T1,T2>&p){os<<p.first<<" "<<p.second;return os;}template<typename T>ostream &operator<<(ostream &os,const vector<T>&v){for(int i=0;i<(int)v.size();i++) os<<v[i]<<(i+1==v.size()?"":" ");return os;}struct UnionFind{vector<int> par;vector<int> siz;int c;//親の個数(グループ数)void init(int n){c=n;for(int i=0;i<n;i++){par.push_back(i);siz.push_back(1);}}UnionFind(int n){init(n);}int root(int x){//根を返すreturn(par[x]==x?x:(par[x]=root(par[x])));}bool sameParent(int x,int y){return(root(x)==root(y));}void unite(int x, int y){x=root(x);y=root(y);if(x==y) return;if(siz[x]<siz[y]) swap(x,y);siz[x] += siz[y];par[y]=x;c--;}int size(int x){return siz[root(x)];}};// u,v,cost,idusing T=tuple<int,int,ll,int>;signed main(){int n,m,k;cin>>n>>m>>k;vector<T>es;ll ans=0;rep(i,m){int u,v;ll c;cin>>u>>v>>c;u--,v--;es.push_back(T(u,v,c,i));}UnionFind uf(n);set<int> banned;rep(i,k){int t;cin>>t;t--;banned.insert(t);auto [u,v,c,_]=es[t];uf.unite(u,v);}sort(ALL(es),[&](T a,T b){return get<2>(a)<get<2>(b);});for(auto [u,v,c,id]:es){if(banned.count(id)) continue;if(uf.sameParent(u,v)){ans+=c;continue;}uf.unite(u,v);}cout<<ans<<endl;return 0;}