HDU2236 Wireless Network【并查集】

题目链接：POJ-2236

kuangbin带你飞【专题五 并查集】

题目描述：

有一个计算机网络的所有线路都坏了，网络中有n台计算机，现在你可以做两种操作，修理（O）和检测两台计算机是否连通（S），只有修理好的计算机才能连通。连通有个规则，两台计算机的距离不能超过给定的最大距离D（一开始会给你n台计算机的坐标）。检测的时候输出两台计算机是否能连通。

思路:

每次修理好一台计算机的时候就遍历一下所有修好的计算机，看距离是否<=D，如果符合说明可以连通，将两台计算机所在集合合并。每次检查的时候判断一下这两台计算机是否在同一集合中即可。

易错点：

1.输出是“FAIL” 不是 “FALL”
2.先判断距离是否满足，然后进行Union操作； 如果先Find根， 然后判断距离会wa掉（这样子就是错误的）。
比如 ：

void Union(int x, int y)
{
	x = Find(x);
	y = Find(y);
	if(x != y)
	{
	    if(pow(p[x].x - p[y].x, 2) + pow(p[x].y - p[y].y, 2) <= d * d)
            p[x].pre = y;
	}
}

这样就会wa掉，原因很简单分析出来！

代码:

#include <iostream>
#include <cstdlib>
#include <cmath>
#include <cstring>
using namespace std;

#define MAX_N 1010

bool vis[MAX_N];//标记vis[k], 表示标号为k电脑的电脑是否可用
int n, d;

struct Node{
	int x, y;
	int pre;//pre 代表祖宗
}p[MAX_N];

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

int Find(int x)
{
	return x == p[x].pre ? x : p[x].pre = Find(p[x].pre);//路径压缩
}

void Union(int x, int y)
{
    x = Find(x);
    y = Find(y);
	if(x != y)
	{
        p[x].pre = y;
	}
}

int main()
{
	cin >> n >> d;
	init(n);
	for(int i = 1; i<=n; i++)
	{
		cin >> p[i].x >> p[i].y;
	}
	memset(vis, 0, sizeof(vis));
	string s;
	while(cin >> s)
	{
		if(s == "O")
		{
			int k;
			cin >> k;
			vis[k] = 1;//标记可用
			for(int i = 1; i<=n; i++)//如果之前有可用 而且距离合适 则进行合并
			{
				if(vis[i] && i != k && pow(p[i].x - p[k].x, 2) + pow(p[i].y - p[k].y, 2) <= d * d)
				{
					Union(i, k);
				}
			}
		}
		else
		{
			int x, y;
			cin >> x >> y;
			if(Find(x) == Find(y))//判断是否是同一个集合
			{
				cout<<"SUCCESS"<<endl;
			}
			else
			{
				cout<<"FAIL"<<endl;
			}
		}
	}
	return 0;
}