poj 2236 Wireless Network

Wireless Network 并查集

题目大意：

有几个点，他们之间不超过一定距离的话，可以联通，
然后有好多点，都是坏的，有两种行为一个是选择修复某一个点，另一个是测试两个点之间是否可以联通，如果可以联通的话，就输出SUCCESS 否则的话就输出FAIL
其中，任意一个点都可以作为中转站

解体思路：

每修复一个点，就看下它与其他已经修复的点直接是否能联通，能得话就联通，
然后构建一个树，判断是否联通就看下根节点是否为同一个根节点

代码：

#include <iostream>
#include <cstring>

struct Greed //node节点
{
    double x, y;
}dir[1111];

int par[1111]; //树
int ok[1111]; //已经修复的点
int rank[1111]; //防止退化

bool vj(Greed a, Greed b, int d) //判断两个点之间是否能联通
{
    if((a.x - b.x) * (a.x - b.x) + (a.y - b.y) * (a.y - b.y) <= d * d) return true;
    return false;
}

int find(int a) //寻找这个点的根节点
{
    if(a == par[a]) return a;
    else return par[a] = find(par[a]);
}

void link(int a, int b) //连接两个点
{
    a = find(a);
    b = find(b);
    if(rank[a] < rank[b]) par[a] = b;
    else par[b] = a;
    if(rank[a] == rank[b]) rank[a]++; //防止树的高度过长
}

int main()
{
    int n, d;
    while(std::cin >> n >> d) //d为最长联通距离
    {
	int cnt = 0;
	memset(rank, 0, sizeof rank);
	for(int i = 1; i < 1111; ++i) par[i] = i; //初始化树
	for(int i = 0; i < n; ++i)
	    std::cin >> dir[i+1].x >> dir[i+1].y;
	char ch;
	while(std::cin >> ch)
	{
	if(ch == 'O') //如果是O，就表示这个点修复了
	{
	    int temp;
	    std::cin >> temp;
	    for(int i = 0; i < cnt; ++i)
	    {
		if(vj(dir[temp], dir[ok[i]], d)) //link 修复的点之间，如果可以联通就联通，构建树
		{
		    link(temp, ok[i]);
		}
	    }
	    ok[cnt++] = temp; //将已经修复的点放数组里
	}
	else //如果是测试的话
	{
	    int a, b;
	    std::cin >> a >> b;
	    //因为任何一个点都可以作为中间节点，所以只需要判断是否有同一根就好了
	    if(find(a) == find(b)) std::cout << "SUCCESS" << std::endl;
	    else std::cout << "FAIL" << std::endl;
	}
	}
    }
    return 0;
}