PAT甲1118 Birds in Forest（25 分）

#include <stdio.h>
#include <string.h>
using namespace std;

int father[10010];
int A[10010];
int N,K,M;
bool vis[10010];
bool vist[10010];

int findfather(int x)
{
    int a=x;
    while(x!=father[x])
    {
        x=father[x];
    }
    while(a!=father[a])
    {
        int z=a;
        a=father[a];
        father[z]=x;
    }
    return x;
}

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


int main()
{
    scanf("%d",&N);
    memset(vis,false,sizeof(vis));
    memset(vist,false,sizeof(vist));
    for(int i=1;i<10010;i++)
    {
        father[i]=i;
    }
    int numb=0,numt=0;
    for(int i=0;i<N;i++)
    {
        int st,b;
        scanf("%d%d",&K,&st);
        if(vis[st]==false)
        {
            numb++;
            vis[st]=true;
        }
        for(int i=0;i<K-1;i++)
        {
            scanf("%d",&b);
            if(vis[b]==false)
            {
                vis[b]=true;
                numb++;
            }
            Union(st,b);
        }
    }
    for(int i=1;i<10010;i++)
    {
        if(vis[i]==true)
        {
            int fa=findfather(i);
            if(vist[fa]==false)
            {
                numt++;
                vist[fa]=true;
            }
        }
    }
    printf("%d %d\n",numt,numb);
    scanf("%d",&M);
    for(int i=0;i<M;i++)
    {
        int c1,c2;
        scanf("%d%d",&c1,&c2);
        int f1=findfather(c1);
        int f2=findfather(c2);
        if(f1!=f2)
        {
            printf("No\n");
        }
        else
        {
            printf("Yes\n");
        }
    }
    return 0;
}