結果

問題 No.748 yuki国のお財布事情
コンテスト
ユーザー amidtentative
提出日時 2020-03-28 19:03:15
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 146 ms / 2,000 ms
コード長 3,592 bytes
コンパイル時間 1,211 ms
コンパイル使用メモリ 103,444 KB
実行使用メモリ 11,844 KB
最終ジャッジ日時 2025-01-02 12:00:54
合計ジャッジ時間 5,056 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
#include <cstring>
#include <complex>
#include <stack>
#include <queue>
#include <unordered_map>
#include <set>
#include <map>

//#define INF 100000000
//#define MOD (int) (1e9+7)
#define rep(i, a) for (int i = 0; i < (a); i++)
using namespace std;

const int MAX = 1000000;
const int MOD = 1000000007;

long long fac[MAX], finv[MAX], inv[MAX];

//テーブルの前処理
void COMinit() {
    fac[0] = fac[1] = 1;
    finv[0] = finv[1] = 1;
    inv[1] = 1;
    for (int i = 2; i < MAX; i++){
        fac[i] = fac[i - 1] * i % MOD;
        //そのまま計算すると負の値になるのでMODを足す
        inv[i] = MOD - inv[MOD%i] * (MOD / i) % MOD;
        finv[i] = finv[i - 1] * inv[i] % MOD;
    }
}

// 二項係数計算
long long COM(int n, int k){
    if (n < k) return 0;
    if (n < 0 || k < 0) return 0;
    return fac[n] * (finv[k] * finv[n - k] % MOD) % MOD;
}

//UnionFind木
struct UnionFind{
   vector<long long int> par; //par[i] : iの親の番号
   vector<long long int> rnk; //root[i] : iの木のサイズ

   //コンストラクタ
   UnionFind(long long int n): par(n), rnk(n){
       //木の初期化
       for(long long int i = 0; i < n; i++){
           par[i] = i;
           rnk[i] = 1;
       }
   }

   //データxが属する木の根を再帰で取得
   long long int root(long long int x){
       if(par[x] == x){
           return x;
       }else{
           return par[x] = root(par[x]);
       }
   }

   //xとyが同じ木に属しているかを判定
   bool same(long long int x, long long int y){
       return root(x) == root(y);
   }

   //xとyの木を併合
   void unite(long long int x, long long int y){
       long long int rx = root(x); //xの根
       long long int ry = root(y); //yの根
       if(rx == ry) return; //根が同じならそのまま

       //併合するときにサイズを足し合わせる
       if(rnk[rx] < rnk[ry]){
           par[rx] = ry;
           rnk[ry] += rnk[rx];
       }else{
           par[ry] = rx;
           rnk[rx] += rnk[ry];
       }
   }

   //属している木のサイズを取得
   long long int size(long long int x){
       return rnk[root(x)];
   }
};


int main(){
    long long int n, m, k;
    cin >> n >> m >> k;

    vector<pair<long long int, pair<long long int, long long int> > > v;

    long long int sum = 0;
    for(long long int i = 0; i < m; i++){
        long long int a, b, c;
        cin >> a >> b >> c;
        sum += c;
        v.push_back(make_pair(c, make_pair(a-1, b-1)));
    }

    //その道が削除できない道か
    bool used[m];
    for(int i = 0; i < m; i++){
        used[i] = false;
    }
    for(long long int i = 0; i < k; i++){
        long long int e;
        cin >> e;
        used[e-1] = true;
    }

    long long int ans = 0;
    vector<pair<long long int, pair<long long int, long long int> > > v_new;
    UnionFind uf(n);
    for(long long int i = 0; i < m; i++){
        if(used[i]){
            int a, b, c;
            a = v[i].second.first;
            b = v[i].second.second;
            c = v[i].first;
            uf.unite(a, b);
            ans += c;
        }else{
            v_new.push_back(v[i]);
        }
    }

    sort(v_new.begin(), v_new.end());
    for(int i = 0; i < v_new.size(); i++){
        int a, b, c;
        a = v_new[i].second.first;
        b = v_new[i].second.second;
        c = v_new[i].first;

        if(uf.same(a, b)) continue;

        uf.unite(a, b);
        ans += c;
    }

    cout << sum - ans << endl;
}
0