結果

問題 No.748 yuki国のお財布事情
コンテスト
ユーザー koyopro
提出日時 2020-05-11 21:25:47
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 52 ms / 2,000 ms
コード長 2,189 bytes
コンパイル時間 1,762 ms
コンパイル使用メモリ 168,460 KB
実行使用メモリ 7,400 KB
最終ジャッジ日時 2024-07-19 05:35:44
合計ジャッジ時間 4,150 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
using namespace std;
#define int long long
#define FOR(i, a, b) for(int i=(a);i<(b);i++)
#define RFOR(i, a, b) for(int i=(b-1);i>=(a);i--)
#define REP(i, n) for(int i=0; i<(n); i++)
#define RREP(i, n) for(int i=(n-1); i>=0; i--)
#define REP1(i, n) for(int i=1; i<=(n); i++)
#define RREP1(i, n) for(int i=(n); i>=1; i--)
#define ALL(a) (a).begin(),(a).end()
#define UNIQUE_SORT(l) sort(ALL(l)); l.erase(unique(ALL(l)), l.end());
#define CONTAIN(a, b) find(ALL(a), (b)) != (a).end()
#define out(...) printf(__VA_ARGS__)

int dxy[] = {0, 1, 0, -1, 0};

void solve();
signed main()
{
#if DEBUG
    std::ifstream in("input.txt");
    std::cin.rdbuf(in.rdbuf());
#endif
    cin.tie(0);
    ios::sync_with_stdio(false);
    solve();
    return 0;
}

/*================================*/
#if DEBUG
#define SIZE 10
#else
#define SIZE 123450
#endif

int N,M,K;

struct UnionFind {
    vector<int> parent;
    int size;
    UnionFind(int size):size(size) {
        reset(size);
    }
    void reset(int size) {
        this->size = size;
        parent.resize(size);
        REP(i,size) parent[i]=i;
    }
    int root(int i) {
        if (parent[i] == i) return i;
        return parent[i] = root(parent[i]);
    }
    void merge(int i, int j) {
        int ir = root(i);
        int jr = root(j);
        parent[ir] = parent[jr] = min(ir, jr);
    }
    bool same(int i, int j) {
        return root(i) == root(j);
    }
};

struct Edge {
    int a,b,c;
    bool used=false;
    
    bool operator < (const Edge &e) const {
        return c < e.c;
    }
};
vector<Edge> E;

void solve() {
    cin>>N>>M>>K;
    REP(i,M) {
        auto e = Edge();
        cin>>e.a>>e.b>>e.c;
        e.a--;e.b--;
        E.push_back(e);
    }
    UnionFind uf = UnionFind(N);
    int j;
    REP(i,K) {
        cin>>j;
        auto e = E[--j];
        uf.merge(e.a, e.b);
        E[j].used=true;
    }
    sort(ALL(E));
    REP(i,M) {
        auto e = E[i];
        if (E[i].used) continue;
        if (!uf.same(e.a, e.b)) {
            uf.merge(e.a, e.b);
            E[i].used=true;
        }
    }
    int ans = 0;
    REP(i,M) if (!E[i].used) ans += E[i].c;
    cout << ans << endl;
}
0