HDU - 1856 More is better （并查集）

HDU - 1856 More is better （并查集）

HDU - 1856

即找元素最多的集合。
设置一个siz[i]数组，表示当前根为 i 的集合元素个数。在Union函数中每次两个集合合并时更新siz[i]。

我的代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn = 10000005;
int a, b;
int f[maxn], siz[maxn];
int ans;

int find(int x){
    if(x == f[x]) return x;
    else return f[x] = find(f[x]);
}
void Union(int a, int b){
    int fa, fb;
    fa = find(a);
    fb = find(b);
    if(fa != fb) {
        f[fa] = fb;
        siz[fb] += siz[fa]; //核心
        ans = max(ans, siz[fb]);
    }
}
void init(){
    for(int i = 1; i <= maxn; i++)
        f[i] = i, siz[i] = 1;//注意siz[i]初始化为1
}
int main(){
    ios::sync_with_stdio(false);cin.tie(0);
    int t;
    while(cin >> t){
        init();
        if(t == 0) {cout << 1 << endl; continue; }//特判一下
        ans = 0;
        while(t--){
            cin >> a >> b;
            Union(a, b);
        }
        cout << ans << endl;
    }
}

WA点：n = 0时，ans = 1.
刚开始直接给ans赋初值1，但是这样输入n=0时还是会进入内循环，没反应并且交上去RE，所以后面特判了。