結果

問題 No.748 yuki国のお財布事情
ユーザー paruki
提出日時 2018-11-04 15:34:59
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 60 ms / 2,000 ms
コード長 2,438 bytes
コンパイル時間 1,928 ms
コンパイル使用メモリ 183,556 KB
実行使用メモリ 10,476 KB
最終ジャッジ日時 2024-11-20 19:57:28
合計ジャッジ時間 3,711 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define _USE_MATH_DEFINES
#include "bits/stdc++.h"
using namespace std;
#define FOR(i,j,k) for(int (i)=(j);(i)<(int)(k);++(i))
#define rep(i,j) FOR(i,0,j)
#define each(x,y) for(auto &(x):(y))
#define mp make_pair
#define MT make_tuple
#define all(x) (x).begin(),(x).end()
#define debug(x) cout<<#x<<": "<<(x)<<endl
#define smax(x,y) (x)=max((x),(y))
#define smin(x,y) (x)=min((x),(y))
#define MEM(x,y) memset((x),(y),sizeof (x))
#define sz(x) (int)(x).size()
#define RT return
using ll = long long;
using pii = pair<int, int>;
using vi = vector<int>;
using vll = vector<ll>;

class UnionFind {
    int cnt;
    vector<int> par, rank, size;
public:
    UnionFind() {}
    UnionFind(int _n) :cnt(_n), par(_n), rank(_n), size(_n, 1) {
        for (int i = 0; i<_n; ++i) par[i] = i;
    }
    int find(int k) {
        return (k == par[k]) ? k : (par[k] = find(par[k]));
    }
    int operator[](int k) {
        return find(k);
    }
    int getSize(int k) {
        return size[find(k)];
    }
    void unite(int x, int y) {
        x = find(x); y = find(y);
        if (x == y) return;
        --cnt;
        if (rank[x] < rank[y]) {
            par[x] = y;
            size[y] += size[x];
        } else {
            par[y] = x;
            size[x] += size[y];
            if (rank[y] == rank[x]) ++rank[x];
        }
    }
    int count() {
        return cnt;
    }
};

template<class Weight>
struct KruskalMST {
    using Edges = vector<tuple<Weight, int, int>>;
    Weight sum;
    Edges mst;
    KruskalMST(int V, Edges edges) :sum(0) {
        sort(edges.begin(), edges.end());
        UnionFind uf(V);
        each(e, edges) {
            Weight w;
            int u, v;
            tie(w, u, v) = e;
            if (uf[u] != uf[v]) {
                mst.emplace_back(e);
                uf.unite(u, v);
                sum += w;
            }
        }
    }
};

int N, M, K;
void solve() {
    cin >> N >> M >> K;
    
    KruskalMST<ll>::Edges edges;
    ll ans = 0;
    rep(i, M) {
        int a, b, c;
        cin >> a >> b >> c;
        --a; --b;
        edges.emplace_back(c, a, b);
        ans += c;
    }

    rep(i, K) {
        int e;
        cin >> e;
        --e;
        ans -= get<0>(edges[e]);
        get<0>(edges[e]) = 0;
    }

    ans -= KruskalMST<ll>(N, edges).sum;
    cout << ans << endl;
}

int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout << fixed << setprecision(15);
	solve();
	return 0;
}
0