L2-024. 部落（并查集）*

L2-024. 部落

参考博客

 #include<cstdio>
 #include<iostream>
 #include<set>
 #include<algorithm>
 #include<cmath>
 #include<cstring>
 #include<string>
 using namespace std;

 int pre[];
 int f[];

 void init()
 {
     for(int i=;i<;i++)
         pre[i]=i,f[i]=;
 }

 int fin(int x)
 {
     return pre[x]==x?x:pre[x]=fin(pre[x]);
 }

 int main()
 {
     init();
     int n,q,k,a,b;
     cin>>n;
     for(int i=;i<n;i++)
     {
         cin>>k;
         cin>>a;
         f[a]=;
         for(int j=;j<k;j++)
         {
             cin>>b;
             f[b]=;
             int x=fin(a);
             int y=fin(b);
             if(x!=y)
                 pre[x]=y;
         }
     }
     int cnt=,tot=;
     for(int i=;i<;i++)
     {
         if(f[i]==)
         {
             cnt++;
             if(pre[i]==i)
                 tot++;
         }
     }
     printf("%d %d\n",cnt,tot);
     cin>>q;
     for(int i=;i<q;i++)
     {
         cin>>a>>b;
         if(fin(a)==fin(b))
             printf("Y\n");
         else
             printf("N\n");
     }

     return ;
 }