PAT 朋友圈(并查集)

朋友圈

某学校有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

做的第一道并查集问题,AC代码如下:

#include<iostream>
#include<string.h>
#define MAX 30010
using namespace std;
int f[MAX];
int fr[MAX];
int n,m;
void Init()
{
    for(int i = 1;i <= MAX; i++)
        f[i] = i;
}
int getf(int v)
{
    if(f[v] == v)
        return v;
    else
    {
        f[v] = getf(f[v]);
        return f[v];
    }
}
void marge(int v,int u)
{
    int t1,t2;
    t1 = getf(v);
    t2 = getf(u);
    if(t1 != t2)
        f[t2] = t1;
}
int main()
{
    int index[MAX],max = 0;
    memset(index,0,sizeof(index));
    cin >> n >> m;
    Init();
    while(m--)
    {
        int d;
        cin >> d;
        for(int i = 0;i < d; i++)
        {
            cin >> fr[i];
            if(i)
                marge(fr[i-1],fr[i]);
        }
    }
    for(int i = 1;i <= n; i++)
        index[getf(i)]++;
    for(int i = 1;i <= n; i++)
    {
        if(max < index[f[i]])
            max = index[f[i]];
    }
    cout << max << endl;
    return 0;
}

猜你喜欢

转载自blog.csdn.net/xiao__hei__hei/article/details/80302405