【并查集】POJ 2236 Wireless Network

POJ 2236 Wireless Network

题意：

• 就是有N台电脑，编号1…N，然后这些电脑有各自的坐标位置。如果输入‘O’（说起这个‘O’我可真是醉了，它的题面上长得跟‘0’一样，编译器上看着也是一样的，开始一直跑不出来，才发现！哦！原来是‘O’！），代表这台电脑可以联网。如果输入‘S’，那么就需要判断后面输入的两个电脑是不是可以互相连通。两个电脑互相连通的条件是：1. 两台电脑各自都可以联网，而且两台电脑的间距不超过所给D。 2. 两台电脑都可以联网，两台电脑的间距超过了所给的D，但是中间有另一台电脑做媒介使之相连。

思路：

• 所以说怎样的两台电脑才是一个集合内的呢？当然是满足条件的！（好像是句废话）前序输入完成之后，我们将 ‘O’ 后面能够联网的电脑编号存在a[ ]数组里，然后用cnt来存储a[ ]的大小。每输入一个修复好的可以联网的电脑，就跑一边1-cnt的循环，看看是不是满足条件可以与已经联网的电脑连通。
• 这样一来，输入S的时候就只需要看看两个电脑是不是在一个集合内就好了。

CODE

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <cmath>
#include <cstring>
#include <string>
#include <vector>
#include <set>
#include <stack>
#include <list>
#include <map>
#define INF 0x3f3f3f3f
#define MID l + r >> 1
#define lsn rt << 1
#define rsn rt << 1 | 1
#define Lson lsn, l, mid
#define Rson rsn, mid+1, r
#define QL Lson, ql, qr
#define QR Rson, ql, qr

using namespace std;
typedef long long ll;
const int maxN = 1001 + 5;
const int maxM = 2e5 + 5;

int N, D, cnt = 0;
int a[maxN];
int root[maxN], high[maxN], size[maxN];
void init()
{
    for(int i = 0 ; i <= N ; i ++ )
    {
        root[i] = i;
        high[i] = 0;
        size[i] = 1;
    }
}
int Find(int x) { return root[x] == x ? x : root[x] = Find(root[x]); }
int Same(int x, int y) { return Find(x) == Find(y); }
void Merge(int x, int y)
{
    x = Find(x); y = Find(y);
    if(high[x] < high[y] || size[x] < size[y])
    {
        root[x] = y;
        size[y] += size[x];
    }
    else
    {
        root[y] = x;
        size[x] += size[y];
        if(high[x] == high[y] ) high[x] ++ ;
    }
}

struct node{
    int x, y;
}dot[maxM];

void haddle()
{
    for(int i = 1; i < cnt ; i ++ )
    {
        if(!Same(a[cnt], a[i]) && (dot[a[cnt]].x - dot[a[i]].x)*(dot[a[cnt]].x - dot[a[i]].x) + (dot[a[cnt]].y - dot[a[i]].y)*(dot[a[cnt]].y - dot[a[i]].y) <= D*D)
            Merge(a[cnt], a[i]);
    }
}

int main()
{
    scanf("%d%d", &N, &D); init();
    for(int i = 1; i <= N ; i ++ )
        scanf("%d%d", &dot[i].x, &dot[i].y);
    char str[10];
    while(~scanf("%s", str))
    {
        if(str[0] == 'O')
        {
            scanf("%d", &a[++cnt]);
            haddle();
        }
        else if(str[0] == 'S')
        {
            int aa, bb; scanf("%d%d",&aa, &bb);
            if(Same(bb, aa))
                printf("SUCCESS\n");
            else
                printf("FAIL\n");
        }
    }
    return 0;
}