北大ACM1703——Find them, Catch them~~并查集

题目的意思是：在一个城市，有两个犯罪团伙。输入N（人的个数）和M（信息的个数）。后面M条信息是一个字母加上两个数字i和j，如果字母为D，则说明i和j属于不同的犯罪团伙，如果字母是A，则要你判断i和j的关系，属于同一个还是属于不同的还是不确定。

一开始，觉得这个题目计较棘手，属于D的情况下，是属于不同的犯罪团伙，不知道如何来解决。

后来想到了之前看到的一题，POJ1182食物链，判断有多少条信息是错的。有三个并查集来判断动物的关系。

所以这一题，我们可以用两个并查集来判断。N最大为10^5，所以并查集要开到2 * 10 ^ 5。i + N，表示 i 属于团伙N。这样就好解决了。

下面是AC的代码：

#include <iostream>
#include <cstdio>
using namespace std;
const int MAX = 100005 * 2;
int par[MAX];
int N, M;

int finds(int x)              //并查集查找函数
{
	if(x == par[x])
		return x;
	else
		return par[x] = finds(par[x]);
}
void join(int x, int y)       //并查集合并函数
{
	x = finds(x);
	y = finds(y);
	if(x != y)
		par[y] = x;
}
int same(int x, int y)        //判断是否是一个并查集
{
	return finds(x) == finds(y);
}

int main()
{
	int i, t, a, b;
	char str;
	scanf("%d", &t);
	while(t--)
	{
		scanf("%d%d", &N, &M);
		getchar();
		for(i = 0; i <= MAX; i++)  //初始化并查集
			par[i] = i;
		for(i = 0; i < M; i++)
		{
			scanf("%c %d %d", &str, &a, &b);
			getchar();
			if(str == 'A')         //输出信息
			{
				if(same(a, b))     //属于同一个
					printf("In the same gang.\n");
				else if(same(a, b + N))   //属于不同的
					printf("In different gangs.\n");
				else                       //不确定的
					printf("Not sure yet.\n");
			}
			else
			{
				join(a, b + N);         //因为是不属于同一个，a与b + N属于同一个团伙
				join(a + N, b);         //a + N 跟b属于一个团伙
			}
		}
	}
	return 0;
}