poj2236（并查集）

题目链接：http://poj.org/problem?id=2236

题意：

　　关于电脑之间的连接问题，给出一个距离限制，一台电脑与一台电脑是连接的有两种情况，一种是他们之间的距离小于限制距离，另一种是他们之间的距离超过了限制距离，但是他们间接通过其他在限制距离内的电脑互相连接。当然要连接的话首先要保证电脑是激活的状态，然后S表示询问，叫你判断询问的两台电脑之间是否连接。

思路：

　　简单的并查集做法，只是合并的时候加上判断条件，首先在距离内的才能直接连接，还有只有连接的两台电脑都处于激活状态。水~

代码：

#include<iostream>
#include<stdio.h>
#include<cmath>
using namespace std;

int fa[1010];
int n, max_dis;

struct Num
{
    int id;
    int x;
    int y;
    int flag;    //判断是否修复
} num[1010];

int dis(Num a, Num b)    //计算距离，返回的是距离的平方
{
    return ((a.x - b.x) * (a.x - b.x) + (a.y - b.y) * (a.y - b.y));
}

void init(int n)
{
    for(int i = 1; i <= n; i++)
    {
        num[i].id = i;
        fa[i] = i;
        num[i].flag = 0;
    }
    return;
}

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

void unite(Num a, Num b)
{
    int r1 = find(a.id);
    int r2 = find(b.id);
    if(r1 == r2)
    {
        return;
    }
    else
    {
        if(dis(a, b) <= (max_dis * max_dis))
        {
            fa[r2] = r1;
        }
    }
}

int main()
{
    ios::sync_with_stdio(false);
    while(cin >> n >> max_dis)
    {
        init(n);
        for(int i = 1; i <= n; i++)
        {
            cin >> num[i].x >> num[i].y;
        }
        char ch;
        int p, q;
        //getchar();
        while(cin >> ch)
        {
            if(ch == 'O')    //这里是大写字母O，不是0
            {
                cin >> p;
                num[p].flag = 1;
                for(int i = 1; i <= n; i++)
                {
                    if(num[i].flag == 1 && i != p)
                        unite(num[p], num[i]);
                }
            }
            else if(ch == 'S')
            {
                cin >> p >> q;
                if(find(num[p].id) == find(num[q].id))
                {
                    cout << "SUCCESS" << endl;
                }
                else
                {
                    cout << "FAIL" << endl;    //最近有点不在状态啊，找了这么久的错误，才发现是FAIL写成了FALL……
                }
            }
        }
    }
    return 0;
}