結果

問題 No.748 yuki国のお財布事情
ユーザー roarisroaris
提出日時 2019-11-21 19:42:35
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 46 ms / 2,000 ms
コード長 1,967 bytes
コンパイル時間 1,878 ms
コンパイル使用メモリ 180,988 KB
実行使用メモリ 11,064 KB
最終ジャッジ日時 2024-10-10 22:42:52
合計ジャッジ時間 3,637 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 2 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 2 ms
5,248 KB
testcase_06 AC 2 ms
5,248 KB
testcase_07 AC 2 ms
5,248 KB
testcase_08 AC 2 ms
5,248 KB
testcase_09 AC 2 ms
5,248 KB
testcase_10 AC 2 ms
5,248 KB
testcase_11 AC 2 ms
5,248 KB
testcase_12 AC 2 ms
5,248 KB
testcase_13 AC 8 ms
5,248 KB
testcase_14 AC 11 ms
5,248 KB
testcase_15 AC 9 ms
5,248 KB
testcase_16 AC 18 ms
5,248 KB
testcase_17 AC 42 ms
9,680 KB
testcase_18 AC 42 ms
8,996 KB
testcase_19 AC 40 ms
8,288 KB
testcase_20 AC 37 ms
7,680 KB
testcase_21 AC 42 ms
9,364 KB
testcase_22 AC 1 ms
5,248 KB
testcase_23 AC 2 ms
5,248 KB
testcase_24 AC 2 ms
5,248 KB
testcase_25 AC 39 ms
9,484 KB
testcase_26 AC 46 ms
11,064 KB
testcase_27 AC 45 ms
11,028 KB
testcase_28 AC 28 ms
6,016 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function 'int main()':
main.cpp:85:16: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17' [-Wc++17-extensions]
   85 |     for (auto& [ai, bi, ci] : edges) {
      |                ^

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define int long long
typedef tuple<int, int, int> T;

struct Unionfind {
    vector<int> par, rank;
    
    Unionfind(int n) {
        par.resize(n);
        fill(par.begin(), par.end(), -1);
        rank.resize(n);
        fill(rank.begin(), rank.end(), 0);
    }
    
    int root(int x) {
        if (par[x]<0) return x;
        par[x] = root(par[x]);
        return par[x];
    }
    
    void unite(int x, int y) {
        int rx = root(x), ry = root(y);
        if (rx==ry) return;
        
        if (rank[rx]>=rank[ry]) {
            par[rx] += par[ry];
            par[ry] = rx;
            
            if (rank[rx]==rank[ry]) {
                rank[rx]++;
            }
        }
        else {
            par[ry] += par[rx];
            par[rx] = ry;
        }
    }
    
    int is_same(int x, int y) {
        return root(x)==root(y);
    }
    
    int count(int x) {
        return -par[root(x)];
    }
};

signed main() {
    cin.tie(0); ios::sync_with_stdio(false);
    
    int N, M, K; cin >> N >> M >> K;
    int a[M], b[M], c[M];
    int cost_sum = 0;
    
    for (int i=0; i<M; i++) {
        cin >> a[i] >> b[i] >> c[i];
        cost_sum += c[i];
    }
    
    int flag[M];
    fill(flag, flag+M, 0);
    
    for (int i=0; i<K; i++) {
        int ei; cin >> ei;
        flag[ei-1] = 1;
    }
    
    vector<T> edges;
    Unionfind uf = Unionfind(N);
    int cost = 0;
    
    for (int i=0; i<M; i++) {
        if (flag[i]) {
            uf.unite(a[i]-1, b[i]-1);
            cost += c[i];
        }
        else {
            edges.push_back(make_tuple(a[i]-1, b[i]-1, c[i]));
        }
    }
    
    sort(edges.begin(), edges.end(), [](T const t1, T const t2) {return get<2>(t1)<get<2>(t2);});
    
    for (auto& [ai, bi, ci] : edges) {
        if (!(uf.is_same(ai, bi))) {
            uf.unite(ai, bi);
            cost += ci;
        }
    }
    
    cout << cost_sum-cost << endl;
}
0