PTA L2-024 部落

题目
复习一下并查集。首先将小部落内的人的father标记为部落序号，接着部落序号进行并查集的操作。

#include <bits/stdc++.h>
using namespace std;
const int N=1e4+10;
int n, tree[N], father[N];
bool vis[N]={false};

/*
简单并查集
int findFather(int x){
    while(x!=father[x]){
        x=father[x];
    }
}
*/

//路径压缩
int findFather(int x){
    if(x==father[x])
        return x;
    else {
        int f=findFather(father[x]);//将某结点到根节点的路径上的结点都进行标记，使其父亲为根节点
        father[x]=f;
        return f;
    }
}

void link(int a, int b){
    int fa=findFather(a);
    int fb=findFather(b);
    if(fa!=fb)
        father[fb]=fa;
}

int main()
{
    int k, a, b, q, maxn=-1;
    scanf("%d", &n);
    for(int i=1; i<=n; i++)
        father[i]=i;

    for(int i=1; i<=n; i++){
        scanf("%d", &k);
        while(k--){
            scanf("%d", &a);
            maxn=max(a, maxn);
            if(!tree[a])
                tree[a]=i;
            else {
                link(tree[a], i);
            }
        }
    }

    int num=0;
    for(int i=1; i<=maxn; i++){
        int f=findFather(tree[i]);
        if(!vis[f]){
            num++;
            vis[f]=true;
        }
    }

    printf("%d %d\n", maxn, num);
    scanf("%d", &q);
    while(q--){
        scanf("%d%d", &a, &b);
        int fa=findFather(tree[a]);
        int fb=findFather(tree[b]);
        if(fa==fb)
            printf("Y\n");
        else
            printf("N\n");
    }
    return 0;
}