数据结构之并查集

1. 有点写复杂了，在find函数里应该返回int而不是data类型比较好

//并查集，在同一类别的指向同一个祖宗，如果两类大小相同，按秩合并(看树的深度),哪个秩小哪个当做根节点
//合并时若秩不相等，则让秩小的合并到秩大的，同时秩不变，如果相等，可以随便合并也可以按下标小的合并，同时祖宗节点的秩加一
//用静态链表存储，每个数据有两个成员，一个是他的祖宗(属于哪一类)，另一个是他属于的那一类的秩
//这里我们采用路径压缩，让一个孩子直接指向他的祖宗，而不是一层一层指向他的组中
#include<stdio.h>
#include<malloc.h>
#include<stdlib.h>


typedef struct data{
	int parent;
	int priority;
}data;

void Init(data a[], int n)
{
	for(int i = 0; i <n;i++)
	{
		a[i].parent = i;
		a[i].priority = 0;
	}
}
// 查找元素所在集合的根节点，同时进行路径压缩优化
data find(data a[], int x) {
    if (a[x].parent!= x) {
        data root = find(a, a[x].parent);
        a[x].parent = root.parent;  // 将当前节点的parent指向找到的根节点，实现路径压缩
        return root;
    }
    return a[x];
}

// 合并两个元素所在的集合
void Merge(data a[], int x, int y) {
    data root_a = find(a, x);
    data root_b = find(a, y);
    if (root_a.parent!= root_b.parent) {  // 确保两个元素不在同一个集合里才进行合并操作
    	//printf("a,b:%d,%d\n", root_a.parent, root_b.parent);
        if (root_a.priority < root_b.priority) {
        {
        	 a[root_a.parent].parent = a[root_b.parent].parent;
        	 //printf("%d\n", root_a.priority);
			}   
        } else if (root_a.priority > root_b.priority) {
            a[root_b.parent].parent = a[root_a.parent].parent;
        } else {
            if (root_a.parent < root_b.parent) {
                a[root_b.parent].parent = a[root_a.parent].parent;
                //printf("root_b.parent:%d", root_b.parent);
            } else {
                a[root_a.parent].parent = a[root_b.parent].parent;
            }
            a[root_a.parent].priority++;
        }
    }
}
data* a;
int main()
{
	printf("输入格式为：先输入有一共有几个，再每一行输入两个数表示在同一类,从零开始:\n");
	int n;
	scanf("%d", &n);
	 a = (data*)malloc(sizeof(data)*(n+1));
	Init(a, n);
	int x,y;
	while(scanf("%d%d", &x, &y) != EOF)
	{
		Merge(a, x, y);
	}
	printf("输出有几个集合:\n");
	int cnt = 0;
	for(int i = 0; i < n; i++)
	{
		//puts("ok");
		//printf("%d ", a[i].parent);
		if(a[i].parent == i)
			cnt++;
	}
	printf("%d", cnt);
	free(a);
	return 0;
}