結果

問題 No.748 yuki国のお財布事情
コンテスト
ユーザー fumiphys
提出日時 2019-03-14 22:59:38
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 55 ms / 2,000 ms
コード長 2,604 bytes
コンパイル時間 1,241 ms
コンパイル使用メモリ 120,756 KB
実行使用メモリ 8,112 KB
最終ジャッジ日時 2024-06-27 07:19:06
合計ジャッジ時間 3,579 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

// includes
#include <cstdio>
#include <cstdint>
#include <iostream>
#include <iomanip>
#include <string>
#include <queue>
#include <stack>
#include <vector>
#include <set>
#include <map>
#include <unordered_map>
#include <algorithm>
#include <utility>
#include <functional>
#include <cmath>
#include <climits>
#include <bitset>
#include <list>
#include <random>

// macros
#define ll long long int
#define pb emplace_back
#define mk make_pair
#define pq priority_queue
#define FOR(i, a, b) for(int i=(a);i<(b);++i)
#define rep(i, n) FOR(i, 0, n)
#define rrep(i, n) for(int i=((int)(n)-1);i>=0;i--)
#define all(x) (x).begin(),(x).end()
#define sz(x) ((int)(x).size())
#define UNIQUE(v) v.erase(unique(v.begin(), v.end()), v.end())

using namespace std;

//  types
typedef pair<int, int> P;
typedef pair<ll, int> Pl;
typedef pair<ll, ll> Pll;
typedef pair<double, double> Pd;
 
// constants
const int inf = 1e9;
const ll linf = 1LL << 50;
const double EPS = 1e-10;
const int mod = 1e9 + 7;

// solve
template <class T>bool chmax(T &a, const T &b){if(a < b){a = b; return 1;} return 0;}
template <class T>bool chmin(T &a, const T &b){if(a > b){a = b; return 1;} return 0;}

vector<pair<int, ll> > edge[100001];

typedef struct UnionFind_ {
  vector<int> par;
  vector<int> rank_;
  UnionFind_(int n): rank_(n, 0) {
    for(int i = 0; i < n; i++)par.push_back(i);
  }
  int find(int x) {
    if(par[x] == x)return x;
    else return par[x] = find(par[x]);
  }
  bool same(int x, int y) {
    if(find(x) == find(y))return true;
    else return false;
  }
  bool unite(int x, int y){
    int xp = find(x);
    int yp = find(y);
    if(xp == yp)return false;
    if(rank_[xp] > rank_[yp])par[yp] = xp;
    else if(rank_[xp] < rank_[yp])par[xp] = yp;
    else {
      par[yp] = xp;
      rank_[xp]++;
    }
    return true;
  }
} UnionFind;

struct ed{
  int a, b;
  ll c;
  ed(){}
  ed(int a, int b, ll c): a(a), b(b), c(c){}
};

bool comp(const ed &a, const ed &b){
  return a.c < b.c;
}

int main(int argc, char const* argv[])
{
  ios_base::sync_with_stdio(false);
  cin.tie(0);
  int n, m, k;
  cin >> n >> m >> k;
  vector<ed> vec(m);
  ll all = 0;
  rep(i, m){
    int a, b;
    ll c;
    cin >> a >> b >> c, a--, b--;
    vec[i] = ed(a, b, c);
    all += c;
  }
  ll res = 0;
  UnionFind uf(n);
  rep(i, k){
    int e;
    cin >> e;
    e--;
    res += vec[e].c;
    uf.unite(vec[e].a, vec[e].b);
  }
  sort(all(vec), comp);
  for(int i = 0; i < m; i++){
    if(!uf.same(vec[i].a, vec[i].b)){
      uf.unite(vec[i].a, vec[i].b);
      res += vec[i].c;
    }
  }
  cout << all - res << endl;
	return 0;
}
0