PAT 7-25 朋友圈

问题描述

某学校有N个学生，形成M个俱乐部。每个俱乐部里的学生有着一定相似的兴趣爱好，形成一个朋友圈。一个学生可以同时属于若干个不同的俱乐部。根据“我的朋友的朋友也是我的朋友”这个推论可以得出，如果A和B是朋友，且B和C是朋友，则A和C也是朋友。请编写程序计算最大朋友圈中有多少人。

输入格式　　
　　输入的第一行包含两个正整数N（≤30000）和M（≤1000），分别代表学校的学生总数和俱乐部的个数。后面的M行每行按以下格式给出1个俱乐部的信息，其中学生从1~N编号：
第i个俱乐部的人数Mi（空格）学生1（空格）学生2 … 学生Mi
输出格式
　　输出给出一个整数，表示在最大朋友圈中有多少人。

输入样例

7 4
3 1 2 3
2 1 4
3 5 6 7
1 6

输出样例

4

解题思路

题目将不同俱乐部的同学划分为不同的朋友圈，通过一位同学在不同俱乐部让大家成为朋友。第一眼看到就意识到使用并查集，但在使用过程中忽略了根的意义，而使用俱乐部编号作为f[i]。

#include<iostream>
#include<algorithm>
using namespace std;
int main()
{
     int n,m;
     //N学校总数，M俱乐部个数 
     cin>>n>>m;
     int a[n],b[m];
     for(int i=0;i<n;i++)
        a[i]=-1;
     for(int i=0;i<m;i++)
        b[i]=0;
     int p;
     for(int i=0;i<m;i++)
     {
          cin>>p;
          int flag=0;
          int ff=0;
          for(int j=1;j<=p;j++)
          {
               int temp;
               cin>>temp;
               //还没有俱乐部 
               if(a[temp-1]==-1)
                    a[temp-1]=i;     
               else
               {
                    flag=1 ; 
                    ff=a[temp-1]; 
               }     
          }
          //该俱乐部有一人参加过别的俱乐部，则归为朋友圈 
          if(flag==1)
          {
               for(int j=0;j<n;j++)
                if(a[j]==i)    
                     a[j]=ff;      
          } 
     }
     for(int z=0;z<n;z++)
      b[a[z]]++;  
     sort(b,b+m);
     cout<<b[m-1];
     return 0; 
}

这样只能通过前两个测试点。
　　本题套用并查集模板即可，将认识的人连起来，最后计算根节点最多的即可。
　　参考链接

解题代码

#include<iostream>
#include<cstring>
using namespace std;
int f[30005];
int cnt[30005];
int find(int a)
{
     if(f[a]==a) return a;
     else return f[a]=find(f[a]);
}
void mer(int a,int b)
{
     int ax=find(a),bx=find(b);
     if(ax!=bx) f[ax]=bx;
}
int main()
{
     int n,m;
     cin>>n>>m;
     memset(cnt,0,sizeof(cnt));
     for(int i=1;i<=n;i++)
         f[i]=i;
     for(int i=0;i<m;i++)
     {
          int s,e,b;
          cin>>s>>b;
          for(int j=1;j<s;j++)
          {
               cin>>e;
               mer(b,e);
          }
     }
     int ans=0;
     for(int i=1;i<=n;i++)
     {
          int y=find(i);
          cnt[y]++;
          if(ans<cnt[y]) ans=cnt[y];
     }
     cout<<ans;
     return 0;
}