并查集

关于并查集的一个代码讲解

该文章借鉴俩位大佬的代码
大佬

大佬

题目：

4-3-1 并查集 朋友圈 (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

大佬的代码：

#include<stdio.h>
int pre[30001];
int sum[30001];
int ans=0;
void init(int n)
{
	for(int i=1;i<=n;i++)
	{
	  pre[i]=i;
	  sum[i]=1;
	}
	  
}
int find(int x)
{
	if(pre[x]==x)return x;
    return pre[x]=find(pre[x]);//路径压缩  
}
void join(int x,int y)
{
	int fx=find(x);
	int fy=find(y);
	if(fx!=fy)
	{
		pre[fx]=fy;
		sum[fy]+=sum[fx];
	}
}
int main()
{
	int n,m;
	scanf("%d %d",&n,&m);
	init(n);
	int s;
	while(m--)
    {
    	scanf("%d",&s);
    	int stus[s+1];
		for(int i=1;i<=s;i++)
		{
	     	scanf("%d",&stus[i]);	
		}
	    for(int x=1;x<=s;x++)
	    {
	    	for(int z=x+1;z<=s;z++)
	    	{
	    		join(stus[x],stus[z]);
			}
		}
	}
	int max=0;
	for(int i=1;i<=n;i++)
	{
		if(sum[i]>max)max=sum[i];
	}
	printf("%d",max);
	return 0;
}

这个代码不长，但是里面有几个我一开始难以理解以及关键的函数：

1.init函数

void init(int n)
{
	for(int i=1;i<=n;i++)
	{
	  pre[i]=i;
	  sum[i]=1;
	}
	  
}

该函数的作用将pre和sum的全局数组初始化，pre函数代表该与该下标意义相同的同学他的上一个节点，sum函数表示该下标的节点及该下标一下的节点个数。
因为初始化，所以sum全部为自己的一个节点，pre全部暂时存为自己下标的节点。

2.find函数

int find(int x)
{
	if(pre[x]==x)return x;
    return pre[x]=find(pre[x]);//路径压缩  
}

该函数的功能是找到朋友x的头节点，通过该函数可以一步一步将pre里面的值变成对应下标的最高节点的下标（路径压缩）

3.join函数

void join(int x,int y)
{
	int fx=find(x);
	int fy=find(y);
	if(fx!=fy)
	{
		pre[fx]=fy;
		sum[fy]+=sum[fx];
	}
}

该函数的作用是判断俩个的朋友是不是同一个圈里面的人，如果是的话就不用再做变化，如果不是，就将一位朋友的节点变成自己最高节点的上一个节点，并且sum加一

再来一题熟悉一下解题方法

4-3-2 并查集 部落 (25 分)
在一个社区里，每个人都有自己的小圈子，还可能同时属于很多不同的朋友圈。我们认为朋友的朋友都算在一个部落里，于是要请你统计一下，在一个给定社区中，到底有多少个互不相交的部落？并且检查任意两个人是否属于同一个部落。

输入格式：
输入在第一行给出一个正整数N（≤10
​4
​​ ），是已知小圈子的个数。随后N行，每行按下列格式给出一个小圈子里的人：

K P[1] P[2] ⋯ P[K]

其中K是小圈子里的人数，P[i]（i=1,⋯,K）是小圈子里每个人的编号。这里所有人的编号从1开始连续编号，最大编号不会超过10
​4
​​ 。

之后一行给出一个非负整数Q（≤10
​4
​​ ），是查询次数。随后Q行，每行给出一对被查询的人的编号。

输出格式：
首先在一行中输出这个社区的总人数、以及互不相交的部落的个数。随后对每一次查询，如果他们属于同一个部落，则在一行中输出Y，否则输出N。

输入样例：

4
3 10 1 2
2 3 4
4 1 5 7 8
3 9 6 4
2
10 5
3 7

输出样例：

10 2
Y
N

代码：

#include<stdio.h>
#define N 10001
int pre[N];
int sum[N];

//功能函数//
void inis();
void join(int x, int y);
int find(int x);


//主函数//
int main()
{
    int n;
    inis();
    scanf("%d", &n);
    int people = 0;
    int m;
    int i = 0, j = 1;
    for (i = 0; i < n; i++) {
        scanf("%d", &m);
        int zb[100];
        for (j = 1; j <= m; j++) {
            scanf("%d", &zb[j]);
            if (zb[j] > people) {
                people = zb[j];
            }
        }
        int x;
        for (j = 1; j <= m; j++) {
            for (x = j + 1; x <= m; x++) {
                join(zb[j], zb[x]);
            }
        }
    }
    int d = 0;
    for (i = 1; i <= people; i++) {
        if (pre[i] == i) {
            d++;
        }
    }
    printf("%d %d\n", people, d);
    int aim1, aim2;
    int nm;
    scanf("%d", &nm);
    for (i = 0; i < nm; i++) {
        scanf("%d %d", &aim1, &aim2);
        int fx = find(aim1);
        int fy = find(aim2);
        if (fx != fy) {
            printf("N\n");
        }
        else {
            printf("Y\n");
        }
    }
}

void inis() {
    for (int i = 1; i <= N; i++) {
        pre[i] = i;
        sum[i] = 1;
    }
}
int find(int x)
{
    if (pre[x] == x)return x;
    return pre[x] = find(pre[x]); 
}
void join(int x, int y)
{
    int fx = find(x);
    int fy = find(y);
    if (fx != fy)
    {
        pre[fx] = fy;
        sum[fy] += sum[fx];
    }
}