K - Wireless Network(25) (POJ - 2236 ）

南亚发生了一次地震。ACM (Asia Cooperated Medical 亚洲联合医疗队) 已经为膝上型电脑搭建了一个无线网络，但受到了一次不可预知的余震攻击，因此网络中的所有电脑都被破坏了。电脑被逐台修复，网络逐步恢复了工作。由于受到硬件的约束，每台电脑只能与距离它不超过 d 米的其它电脑直接通信。但每台电脑可被看作其它两台电脑的通信中转点，也就是说，如果电脑 A 和电脑 B 可以直接通信，或存在一台电脑 C 既可与 A 也可与 B 通信，那么电脑 A 和电脑 B 之间就能够通信。 

在处理网络修复的过程中，工作人员们在任何一个时刻，可以执行两种操作：维修一台电脑，或测试两台电脑是否能够通信。请您找出全部的测试操作。 

输入

第一行包含了两个整数 N 和 d (1 <= N <= 1001, 0 <= d <= 20000)。此处 N 是电脑的数目，编号从 1 到 N；同时，D 是两台电脑之间能够直接通信的最大距离。接下来的 N 行，每行包含两个整数 xi, yi (0 <= xi, yi <= 10000)，表示 N 台电脑的坐标。从第 (N+1) 行到输入结束，是逐一执行的操作，每行包含一个操作，格式是以下两者之一： 
1. "O p" (1 <= p <= N)，表示维护电脑 p 。 
2. "S p q" (1 <= p, q <= N)，表示测试电脑 p 和 q 是否能够通信。 

输入不超过 300000 行。 

输出

对于每个测试操作，如果两台电脑能够通信，则打印 "SUCCESS"；否则，打印 "FAIL"。

示例输入

4 1
0 1
0 2
0 3
0 4
O 1
O 2
O 4
S 1 4
O 3
S 1 4

示例输出

FAIL
SUCCESS

此题考察并查集算法

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

int f[1010],s[1010];
int n[1010],m[1010];

int getx(int x)
{
    while(x!=f[x])//return x==f[x]?x:getx(f[x]);
    {
        x=f[x];
    }
    return x;
}

void build(int x,int y)
{
    int dx=getx(x);
    int dy=getx(y);
    if(dx!=dy)
    {
        f[dx]=dy;
    }
}

int main()
{
    int a,b,i,x,y;
    char c;
    scanf("%d%d",&a,&b);
    memset(f,0,sizeof(f));
    for(i=1; i<=a; i++)
    {
        scanf("%d%d",&n[i],&m[i]);
        f[i]=i;
        s[i]=0;
    }
    while(cin >> c)//必须要用这样输入不然会超时
    {
        if(c=='O')
        {
            scanf("%d",&x);
            s[x]=1;
            for(i=1; i<=a; i++)
            {
                if(i!=x&&s[i])
                {
                    if((m[x]-m[i])*(m[x]-m[i])+(n[x]-n[i])*(n[x]-n[i])<=b*b)
                    {
                        build(x,i);
                    }
                }
            }
        }
        else if(c=='S')
        {
            scanf("%d%d",&x,&y);
            if(getx(x)==getx(y))
            {
                printf("SUCCESS\n");
            }
            else
            {
                printf("FAIL\n");
            }
        }
    }
    return 0;
}