結果

問題 No.748 yuki国のお財布事情
ユーザー roaris
提出日時 2019-11-21 19:41:42
言語 C++14
(gcc 8.3.0)
結果
WA   .
実行時間 -
コード長 1,942 Byte
コンパイル時間 1,798 ms
使用メモリ 8,924 KB
最終ジャッジ日時 2019-11-21 19:41:47

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
000_sample_001 AC 3 ms
6,872 KB
000_sample_002 AC 4 ms
6,876 KB
000_sample_003 AC 3 ms
6,872 KB
001_rand_001 AC 4 ms
6,876 KB
001_rand_002 AC 4 ms
6,872 KB
001_rand_003 AC 3 ms
6,872 KB
001_rand_004 AC 4 ms
6,872 KB
001_rand_005 AC 2 ms
6,872 KB
001_rand_006 AC 3 ms
6,872 KB
001_rand_007 AC 4 ms
6,876 KB
001_rand_008 AC 4 ms
6,876 KB
001_rand_009 AC 2 ms
6,876 KB
001_rand_010 AC 4 ms
6,876 KB
002_rand_001 WA -
002_rand_002 WA -
002_rand_003 WA -
002_rand_004 WA -
002_rand_005 WA -
002_rand_006 WA -
002_rand_007 WA -
002_rand_008 WA -
002_rand_009 WA -
100_hand_001 AC 4 ms
6,876 KB
100_hand_002 AC 4 ms
6,872 KB
100_hand_003 AC 3 ms
6,876 KB
100_hand_004 WA -
100_hand_005 AC 45 ms
6,872 KB
100_hand_006 WA -
100_hand_007 AC 28 ms
6,872 KB
テストケース一括ダウンロード
コンパイルメッセージ
main.cpp: 関数 ‘int main()’ 内:
main.cpp:84:16: 警告: structured bindings only available with -std=c++17 or -std=gnu++17
     for (auto& [ai, bi, ci] : edges) {
                ^

ソースコード

diff #
#include <bits/stdc++.h>
using namespace std;
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)];
    }
};

int 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