結果

問題 No.748 yuki国のお財布事情
ユーザー beetbeet
提出日時 2018-10-19 22:14:08
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 134 ms / 2,000 ms
コード長 1,732 bytes
コンパイル時間 2,337 ms
コンパイル使用メモリ 213,152 KB
実行使用メモリ 13,992 KB
最終ジャッジ日時 2024-11-18 21:01:45
合計ジャッジ時間 4,544 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,816 KB
testcase_02 AC 2 ms
6,816 KB
testcase_03 AC 2 ms
6,816 KB
testcase_04 AC 2 ms
6,816 KB
testcase_05 AC 2 ms
6,816 KB
testcase_06 AC 1 ms
6,816 KB
testcase_07 AC 2 ms
6,820 KB
testcase_08 AC 2 ms
6,816 KB
testcase_09 AC 2 ms
6,816 KB
testcase_10 AC 2 ms
6,816 KB
testcase_11 AC 2 ms
6,816 KB
testcase_12 AC 2 ms
6,816 KB
testcase_13 AC 16 ms
6,820 KB
testcase_14 AC 28 ms
6,816 KB
testcase_15 AC 20 ms
6,816 KB
testcase_16 AC 51 ms
7,440 KB
testcase_17 AC 104 ms
11,656 KB
testcase_18 AC 118 ms
13,124 KB
testcase_19 AC 134 ms
13,992 KB
testcase_20 AC 122 ms
12,932 KB
testcase_21 AC 114 ms
12,408 KB
testcase_22 AC 2 ms
6,816 KB
testcase_23 AC 2 ms
6,816 KB
testcase_24 AC 2 ms
6,816 KB
testcase_25 AC 93 ms
11,892 KB
testcase_26 AC 121 ms
13,144 KB
testcase_27 AC 107 ms
12,308 KB
testcase_28 AC 90 ms
11,152 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using Int = long long;
template<typename T1,typename T2> inline void chmin(T1 &a,T2 b){if(a>b) a=b;}
template<typename T1,typename T2> inline void chmax(T1 &a,T2 b){if(a<b) a=b;}


template<typename T>
struct Kruskal{
  
  struct edge{
    Int from,to;
    T cost;
    Int used;
    edge(){}
    edge(Int from,Int to,T cost):
      from(from),to(to),cost(cost),used(0){}
    bool operator<(const edge& e) const{
      return cost<e.cost;
    }
  };

  Int n;
  vector<Int> p,r;
  vector<edge> edges;

  Kruskal(){}
  Kruskal(Int n):n(n){}

  void init(Int n){
    r.assign(n,1);
    p.resize(n);
    iota(p.begin(),p.end(),0);
  }
  
  Int find(Int x){
    return (x==p[x]?x:p[x]=find(p[x]));
  }

  bool same(Int x,Int y){
    return find(x)==find(y);
  }

  void unite(Int x,Int y){
    x=find(x);y=find(y);
    if(x==y) return;
    if(r[x]<r[y]) swap(x,y);
    r[x]+=r[y];
    p[y]=x;
  }

  void add_edge(Int u,Int v,T c){
    edges.emplace_back(u,v,c);
  }
  
  Int build(){
    sort(edges.begin(),edges.end());
    init(n);
    Int res=0;
    for(auto &e:edges){
      if(!e.cost.first||!same(e.from,e.to)){
	res+=e.cost.second;
	unite(e.from,e.to);
	e.used=1;
      }
    }
    return res;
  }
};

//INSERT ABOVE HERE
signed main(){
  Int n,m,k;
  cin>>n>>m>>k;
  vector<Int> a(m),b(m),c(m);
  Int sum=0;
  for(Int i=0;i<m;i++){
    cin>>a[i]>>b[i]>>c[i];
    a[i]--;b[i]--;
    sum+=c[i];
  }
  vector<Int> e(k);
  for(Int i=0;i<k;i++) cin>>e[i];
  vector<Int> u(m);
  for(Int i=0;i<k;i++) u[--e[i]]=1;

  using P = pair<Int, Int>;
  Kruskal<P> G(n);  
  for(Int i=0;i<m;i++)
    G.add_edge(a[i],b[i],P(!u[i],c[i]));

  Int res=G.build();
  
  cout<<sum-res<<endl;
  return 0;
}
0