并查集(union-find)模板

#include
#include
#include
using namespace std;
const int mx = 100005;

int fa[mx], rk[mx];
vector element[mx];

inline void init(int n)
{
	memset(fa, -1, sizeof(fa)); ///为防止数据中有0号节点,可先将所有的父节点置为-1
	memset(rk, 0, sizeof(rk));
}

int find(int x) {return ~fa[x] ? fa[x] = find(fa[x]) : x;}

void merge(int x, int y)
{
	x = find(x), y = find(y);
	if (x == y) return; ///已合并
	if (rk[x] < rk[y])
		fa[x] = y;
	else
	{
		fa[y] = x;
		if (rk[x] == rk[y]) ++rk[x];
	}
}

inline bool same(int x, int y) {return find(x) == find(y);}

///输出有多少个不同集合
int set_counter(int n)
{
	int sum = 0;
	for (int i = 0; i < n; ++i)
		if (find(i) == -1) ++sum;
	return sum;
}

///输出所有集合
void outputset(int n)
{
	int i, j;
	for (i = 0; i < n; ++i) element[i].clear();
	for (i = 0; i < n; ++i) element[~find(i) ? find(i) : i].push_back(i);
	for (i = 0; i < n; ++i)
		if (element[i].size())
		{
			putchar('{');
			for (j = 0; j < element[i].size() - 1; ++j)
				printf("%d,", element[i][j]); /// 注意数组内元素本身就是有序的
			printf("%d}\n", element[i][element[i].size() - 1]);
		}
}

int main()
{
	int n, m, a, b; ///m是(a,b)的对数
	while (~scanf("%d%d", &n, &m))
	{
		init(n);
		while (m--)
		{
			scanf("%d%d", &a, &b);
			merge(a, b);
		}
		printf("%d\n", set_counter(n)); ///输出有多少个不同集合
		outputset(n);///输出所有集合
	}
	return 0;
}

你可能感兴趣的:(算法详解&模板,acm之路--数据结构,并查集)