結果

問題 No.748 yuki国のお財布事情
ユーザー parukiparuki
提出日時 2018-11-04 15:34:59
言語 C++14
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 56 ms / 2,000 ms
コード長 2,438 bytes
コンパイル時間 1,907 ms
コンパイル使用メモリ 182,224 KB
実行使用メモリ 10,420 KB
最終ジャッジ日時 2023-08-13 03:21:01
合計ジャッジ時間 3,967 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge13
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,376 KB
testcase_01 AC 2 ms
4,380 KB
testcase_02 AC 2 ms
4,376 KB
testcase_03 AC 1 ms
4,376 KB
testcase_04 AC 2 ms
4,380 KB
testcase_05 AC 1 ms
4,380 KB
testcase_06 AC 2 ms
4,384 KB
testcase_07 AC 1 ms
4,384 KB
testcase_08 AC 2 ms
4,376 KB
testcase_09 AC 1 ms
4,376 KB
testcase_10 AC 2 ms
4,384 KB
testcase_11 AC 1 ms
4,376 KB
testcase_12 AC 1 ms
4,380 KB
testcase_13 AC 7 ms
4,380 KB
testcase_14 AC 12 ms
4,376 KB
testcase_15 AC 8 ms
4,380 KB
testcase_16 AC 22 ms
5,360 KB
testcase_17 AC 42 ms
7,252 KB
testcase_18 AC 51 ms
8,908 KB
testcase_19 AC 56 ms
10,420 KB
testcase_20 AC 51 ms
9,908 KB
testcase_21 AC 50 ms
10,092 KB
testcase_22 AC 2 ms
4,376 KB
testcase_23 AC 2 ms
4,380 KB
testcase_24 AC 1 ms
4,376 KB
testcase_25 AC 39 ms
6,132 KB
testcase_26 AC 50 ms
10,360 KB
testcase_27 AC 49 ms
10,356 KB
testcase_28 AC 39 ms
7,996 KB
権限があれば一括ダウンロードができます

ソースコード

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