結果

問題 No.748 yuki国のお財布事情
ユーザー Taiki0715Taiki0715
提出日時 2024-07-31 17:38:11
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 74 ms / 2,000 ms
コード長 2,982 bytes
コンパイル時間 5,655 ms
コンパイル使用メモリ 319,644 KB
実行使用メモリ 12,408 KB
最終ジャッジ日時 2024-07-31 17:38:20
合計ジャッジ時間 8,684 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 2 ms
6,940 KB
testcase_04 AC 2 ms
6,940 KB
testcase_05 AC 2 ms
6,944 KB
testcase_06 AC 2 ms
6,940 KB
testcase_07 AC 1 ms
6,940 KB
testcase_08 AC 2 ms
6,944 KB
testcase_09 AC 2 ms
6,944 KB
testcase_10 AC 2 ms
6,940 KB
testcase_11 AC 2 ms
6,944 KB
testcase_12 AC 2 ms
6,944 KB
testcase_13 AC 6 ms
6,944 KB
testcase_14 AC 10 ms
6,940 KB
testcase_15 AC 7 ms
6,944 KB
testcase_16 AC 18 ms
6,940 KB
testcase_17 AC 39 ms
6,944 KB
testcase_18 AC 46 ms
7,832 KB
testcase_19 AC 51 ms
8,912 KB
testcase_20 AC 45 ms
8,268 KB
testcase_21 AC 61 ms
10,248 KB
testcase_22 AC 2 ms
6,948 KB
testcase_23 AC 2 ms
6,940 KB
testcase_24 AC 1 ms
6,940 KB
testcase_25 AC 30 ms
6,940 KB
testcase_26 AC 67 ms
12,252 KB
testcase_27 AC 74 ms
12,408 KB
testcase_28 AC 29 ms
6,944 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#if __has_include(<atcoder/all>)
#include <atcoder/all>
using namespace atcoder;
template<int mod>istream &operator>>(istream &is,static_modint<mod> &a){long long b;is>>b;a=b;return is;}
istream &operator>>(istream &is,modint &a){long long b;cin>>b;a=b;return is;}
#endif
#ifdef LOCAL
#include "debug.h"
#else
#define debug(...) static_cast<void>(0)
#define debugg(...) static_cast<void>(0)
#define esper(...) static_cast<void>(0)
template<typename T1,typename T2>ostream &operator<<(ostream &os,const pair<T1,T2>&p){os<<p.first<<' '<<p.second;return os;}
#endif
using ll=long long;
using ull=unsigned long long;
using P=pair<ll,ll>;
template<typename T>using minque=priority_queue<T,vector<T>,greater<T>>;
template<typename T>bool chmax(T &a,const T &b){return (a<b?(a=b,true):false);}
template<typename T>bool chmin(T &a,const T &b){return (a>b?(a=b,true):false);}
template<typename T1,typename T2>istream &operator>>(istream &is,pair<T1,T2>&p){is>>p.first>>p.second;return is;}
template<typename T>istream &operator>>(istream &is,vector<T> &a){for(auto &i:a)is>>i;return is;}
template<typename T1,typename T2>void operator++(pair<T1,T2>&a,int n){a.first++,a.second++;}
template<typename T1,typename T2>void operator--(pair<T1,T2>&a,int n){a.first--,a.second--;}
template<typename T>void operator++(vector<T>&a,int n){for(auto &i:a)i++;}
template<typename T>void operator--(vector<T>&a,int n){for(auto &i:a)i--;}
#define reps(i,a,n) for(int i=(a);i<int(n);i++)
#define rep(i,n) reps(i,0,n)
#define all(x) x.begin(),x.end()
#define pcnt(x) __builtin_popcountll(x)
#define fin(x) return cout<<x<<'\n',static_cast<void>(0)
ll myceil(ll a,ll b){return (a+b-1)/b;}
template<typename T,size_t n,size_t id=0>
auto vec(const int (&d)[n],const T &init=T()){
  if constexpr (id<n)return vector(d[id],vec<T,n,id+1>(d,init));
  else return init;
}
void SOLVE();
int main(){
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  cout<<fixed<<setprecision(16);
  #ifdef LOCAL
  clock_t start=clock();
  #endif
  int testcase=1;
  //cin>>testcase;
  for(int i=0;i<testcase;i++){
    SOLVE();
  }
  #ifdef LOCAL
  cerr<<"time:";
  cerr<<(clock()-start)/1000;
  cerr<<"ms\n";
  #endif
}
void SOLVE(){
  int n,m,k;
  cin>>n>>m>>k;
  vector<tuple<int,int,ll>>edge(m);
  ll ans=0;
  rep(i,m){
    int a,b,c;
    cin>>a>>b>>c;
    a--,b--;
    edge[i]={a,b,c};
    ans+=c;
  }
  dsu uf(n);
  while(k--){
    int e;
    cin>>e;
    e--;
    auto [u,v,c]=edge[e];
    uf.merge(u,v);
    ans-=c;
  }
  while(uf.size(0)<n){
    auto g=uf.groups();
    vector<int>b(n);
    rep(i,g.size())for(auto j:g[i])b[j]=i;
    vector<ll>cost(g.size(),2e9),idx(g.size(),-1);
    for(auto [u,v,c]:edge)if(b[u]!=b[v]){
      if(chmin(cost[b[u]],c))idx[b[u]]=v;
      if(chmin(cost[b[v]],c))idx[b[v]]=u;
    }
    rep(i,g.size())if(idx[i]!=-1){
      if(!uf.same(g[i][0],idx[i])){
        uf.merge(g[i][0],idx[i]);
        ans-=cost[i];
      }
    }
  }
  cout<<ans<<endl;
}
0