L2-024 部落 (25分)

社区总人数我是找编号的max（因为1-N编号），然后就普通并查集

测试点4错了，init那个是<=

#include<bits/stdc++.h>
using namespace std;
int father[10010];
void init(){
    for(int i=0;i<=10000;i++) father[i]=i;
    return;
}
int getFather(int a){
    int x=a;
    while(x!=father[x]) x=father[x];
    while(a!=father[a]){
        int z=a;
        father[a]=x;
        a=father[z];
    }
    return x;
}
void unionu(int a,int b){
    int fa=getFather(a);int fb=getFather(b);
    if(fa!=fb) father[fa]=fb;
}
int main(){
	init();
    int n;cin>>n;
    int max=-1;
    for(int i=0;i<n;i++){
        int k;cin>>k;int x,y;cin>>x;
        if(x>max) max=x;
        for(int j=0;j<k-1;j++){
            cin>>y;
            if(y>max) max=y;
            unionu(x,y);
        }
    }
    int m;cin>>m;
    cout<<max<<" ";
    int cnt=0;
    for(int i=1;i<=max;i++){
        getFather(i);
    }
    for(int i=1;i<=max;i++){
        if(i==father[i]) cnt++;
    }
    cout<<cnt<<endl;
    for(int i=0;i<m;i++){
        int x,y;
        cin>>x>>y;
        if(father[x]==father[y]) cout<<"Y"<<endl;
        else cout<<"N"<<endl;
    }
    return 0;
}