14.数据结构之QuickFind并查集

天蝎座的程序员

于 2022-06-30 20:32:24 发布

阅读量340

点赞数

CC 4.0 BY-SA版权

分类专栏：数据结构文章标签：数据结构链表

本文链接：https://blog.csdn.net/xiaoduan6/article/details/125547782

数据结构专栏收录该内容

25 篇文章

订阅专栏

该代码实现了一个并查集的数据结构，包括初始化、查找、合并等操作。在主函数中，通过读取输入进行一系列操作，判断两个元素是否属于同一集合，用于解决集合连接问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

代码

#include <iostream>
#include <stdio.h>
#include <stdlib.h>

typedef struct UnionSet
{
	int* color;
	int size;
}UnionSet;

UnionSet* initSet(int n)
{
	UnionSet* u = (UnionSet*)malloc(sizeof(UnionSet));
	u->color = (int*)malloc(sizeof(int) * n);
	u->size = n;
	for (int i = 0; i < u->size; i++)
	{
		u->color[i] = i;
	}
	return u;
}

void freeSet(UnionSet* u)
{
	if (!u) return;
	free(u->color);
	free(u);
	return;
}


int find(UnionSet* u, int index)
{
	return u->color[index];
}

bool merge(UnionSet* u, int a, int b)
{
	if (find(u, a) == find(u, b)) return false;
	int acolor = find(u, a);
	for (int i = 0; i < u->size; i++)
	{
		if (find(u, i) == acolor)
			u->color[i] = u->color[b];
	}
	return true;
}



int main()
{
	int n, m;
	scanf("%d%d", &n ,& m);

	UnionSet* u = initSet(n);

	for (int i = 0; i < m; i++)
	{
		int a, b, c;
		scanf("%d%d%d", &a, &b, &c);
		switch (a)
		{
		case 1:
			merge(u, b, c);
			break;
		case 2:
			printf("%s\n", find(u, b) == find(u, c) ? "Yes" : "No");
			break;
		}
	}

	freeSet(u);


	return 0;
}