結果

問題 No.556 仁義なきサルたち
ユーザー Manuel1024
提出日時 2021-05-06 11:07:45
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 21 ms / 2,000 ms
コード長 1,227 bytes
コンパイル時間 809 ms
コンパイル使用メモリ 76,340 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-09-14 03:49:35
合計ジャッジ時間 1,901 ms
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

struct UnionFind{
    vector<int> parent;
    UnionFind(int n){
        parent = vector<int>(n, -1);
    }

    int root(int x){
        if(parent[x] < 0) return x;
        else return parent[x] = root(parent[x]);
    }

    int count(int x){
        return -parent[root(x)];
    }

    bool same(int x, int y){
        return root(x) == root(y);
    }

    void unite(int winner, int loser){
        winner = root(winner);
        loser = root(loser);
        parent[winner] += parent[loser];
        parent[loser] = winner;
    }
};

int main(){
    int n, m;
    cin >> n >> m;
    UnionFind data(n);
    auto winner = [&](int x, int y){
        if(data.count(x) == data.count(y)) return min(x, y);
        else if(data.count(x) > data.count(y)) return x;
        else return y;
    };
    for(int i = 0; i < m; i++){
        int a, b;
        cin >> a >> b;
        a--; b--;
        a = data.root(a);
        b = data.root(b);
        if(data.same(a, b)) continue;

        int w = winner(a, b);
        int l = (w == a) ? b : a;
        data.unite(w, l);
    }

    for(int i = 0; i < n; i++) cout << data.root(i)+1 << endl;
    return 0;
}
0