結果

問題 No.1639 最小通信路
ユーザー rieaaddlreiuu
提出日時 2024-05-28 19:06:30
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,905 bytes
コンパイル時間 1,982 ms
コンパイル使用メモリ 174,536 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-12-20 20:51:16
合計ジャッジ時間 9,696 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 1 RE * 1
other RE * 43
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <regex>
#include <algorithm>
using namespace std;
//printf , scanfが使えるよ！！
/*
[問題メモ]


*/

struct UnionFind {
    vector<int> par, rank, siz;

    // 構造体の初期化
    UnionFind(int n) : par(n,-1), rank(n,0), siz(n,1) { }

    // 根を求める
    int root(int x) {
        if (par[x]==-1) return x; // x が根の場合は x を返す
        else return par[x] = root(par[x]); // 経路圧縮
    }

    // x と y が同じグループに属するか (= 根が一致するか)
    bool issame(int x, int y) {
        return root(x)==root(y);
    }

    // x を含むグループと y を含むグループを併合する
    bool unite(int x, int y) {
        int rx = root(x), ry = root(y); // x 側と y 側の根を取得する
        if (rx==ry) return false; // すでに同じグループのときは何もしない
        // union by rank
        if (rank[rx]<rank[ry]) swap(rx, ry); // ry 側の rank が小さくなるようにする
        par[ry] = rx; // ry を rx の子とする
        if (rank[rx]==rank[ry]) rank[rx]++; // rx 側の rank を調整する
        siz[rx] += siz[ry]; // rx 側の siz を調整する
        return true;
    }

    // x を含む根付き木のサイズを求める
    int size(int x) {
        return siz[root(x)];
    }
};

int main() {
    //Let's algorithm!!
    int N,M;
    cin >> N;
    M = N*(N-1)/2;
    int u,v,w;
    UnionFind union_find(N);
    vector<vector<int>> edges(M);
    for(int i=0;i<M;i++){
        cin >> u >> v >> w;
        edges[i].push_back(w);
        edges[i].push_back(u);
        edges[i].push_back(v);
    }
    int min = 0;
    for(int i=0;i<M;i++){
        if(!union_find.issame(edges[i][0],edges[i][1])){
            union_find.unite(edges[i][0],edges[i][1]);
            min += edges[i][2];
        }
    }
    cout << min << endl;
    return 0;
}
0