結果

問題 No.3087 University Coloring
ユーザー yuusaan
提出日時 2025-03-11 13:06:49
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 244 ms / 2,000 ms
コード長 1,878 bytes
コンパイル時間 1,350 ms
コンパイル使用メモリ 115,548 KB
実行使用メモリ 7,328 KB
最終ジャッジ日時 2025-04-04 20:33:34
合計ジャッジ時間 9,018 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff #

//クラスカル法による解法
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;

struct UnionFind {
    vector<int> par; // par[i]:iの親の番号 (例) par[3] = 2 : 3の親が2
    
    UnionFind(int N) : par(N) { //最初は全てが根であるとして初期化
        for(int i = 0; i < N; i++) par[i] = i;
    }
    
    int root(int x) { // データxが属する木の根を再帰で得る:root(x) = {xの木の根}
        if (par[x] == x) return x;
        return par[x] = root(par[x]);
    }
    
    void unite(int x, int y) { // xとyの木を併合
        int rx = root(x); //xの根をrx
        int ry = root(y); //yの根をry
        if (rx == ry) return; //xとyの根が同じ(=同じ木にある)時はそのまま
        par[rx] = ry; //xとyの根が同じでない(=同じ木にない)時:xの根rxをyの根ryにつける
    }
    
    bool same(int x, int y) { // 2つのデータx, yが属する木が同じならtrueを返す
        int rx = root(x);
        int ry = root(y);
        return rx == ry;
    }
};

int main(){
    int N,M;
    cin >> N >> M;
    vector<tuple<long long,int,int>> edge(M);
    for(int i = 0;i < M;i++){
        int a,b;
        long long  c;
        cin >> a >> b >> c;
        a--,b--;
        edge[i] = {c,a,b};
    }
    sort(edge.rbegin(),edge.rend());//辺を長さ順に降順ソートする

    UnionFind uf(N);

    long long ans = 0;
    for(int i = 0;i < M;i++){//長さが長い方から順に見ていく
        int a,b;
        long long c;
        tie(c,a,b) = edge[i];
        if(!uf.same(a,b)){//もし、2頂点が繋がっていないなら、答えに追加して繋ぐ
            ans += c;
            uf.unite(a,b);
        }
    }

    cout << ans * 2 << endl;//答えは最大全域木の長さの総和の2倍であることに注意
    return 0;
}
0