POJ2236 Wireless Network 并查集 好题

　　　　　　　　　　Wireless Network

Description

An earthquake takes place in Southeast Asia. The ACM (Asia Cooperated Medical team) have set up a wireless network with the lap computers, but an unexpected aftershock attacked, all computers in the network were all broken. The computers are repaired one by one, and the network gradually began to work again. Because of the hardware restricts, each computer can only directly communicate with the computers that are not farther than d meters from it. But every computer can be regarded as the intermediary of the communication between two other computers, that is to say computer A and computer B can communicate if computer A and computer B can communicate directly or there is a computer C that can communicate with both A and B. 

In the process of repairing the network, workers can take two kinds of operations at every moment, repairing a computer, or testing if two computers can communicate. Your job is to answer all the testing operations. 

Input

The first line contains two integers N and d (1 <= N <= 1001, 0 <= d <= 20000). Here N is the number of computers, which are numbered from 1 to N, and D is the maximum distance two computers can communicate directly. In the next N lines, each contains two integers xi, yi (0 <= xi, yi <= 10000), which is the coordinate of N computers. From the (N+1)-th line to the end of input, there are operations, which are carried out one by one. Each line contains an operation in one of following two formats: 
1. "O p" (1 <= p <= N), which means repairing computer p. 
2. "S p q" (1 <= p, q <= N), which means testing whether computer p and q can communicate. 

The input will not exceed 300000 lines. 

Output

For each Testing operation, print "SUCCESS" if the two computers can communicate, or "FAIL" if not.

Sample Input

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

Sample Output

FAIL
SUCCESS


题意：现在有n个电脑，按1开始编号，规定2个电脑可以直接通讯的条件：
　　　1.2个电脑都有维修过
　　　2.2个电脑的距离<=d
　　　当然，2个电脑也可以通过其他电脑间接联系

现在给出了n，d，
然后给出n台电脑的坐标(2维)，然后再给出操作：
O a   ：维修电脑a
S a b ：test电脑a和b是否可以连接

思路：
先算好dis[i][j] 表示i与j的距离(这个要先预处理好，不然等下会重复计算很多次，会tle)
O a   则vis[a]=true  然后遍历一遍，把vis[i]==true&&dis[a][i]<=d  的点i 和a 合并
S a b   若a和b的祖先节点相同，则SUCCESS，否则FAIL

学到了，先对数据进行预处理。

#include<cstdio>
#include<cmath>
using namespace std;

const int maxn=1005;
double dis[maxn][maxn];
int father[maxn];
bool vis[maxn];
int n,d;
struct Point
{
    int x,y;
}point[maxn];

void init_distance()
{
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
         dis[i][j]=sqrt((long double)(point[i].x-point[j].x)
*(point[i].x-point[j].x)+(long double)(point[i].y-point[j].y)*(point[i].y-point[j].y));
}

void init()
{
    for(int i=1;i<=n;i++)
    {
        father[i]=i;
        vis[i]=false;
    }
}

int find_set(int x)
{
    if(father[x]==x)
        return x;
    else
        return father[x]=find_set(father[x]);
}

void union_set(int x,int y)
{
    int fax=find_set(x);
    int fay=find_set(y);
    if(fax==fay)
        return ;
    father[fax]=fay;
}

int main()
{
    scanf("%d%d",&n,&d);
    for(int i=1;i<=n;i++)
        scanf("%d%d",&point[i].x,&point[i].y);

    init();
    init_distance();
    char str[3];
    int a,b;
    while(scanf("%s",&str)!=EOF)
    {
        if(str[0]=='O')
        {
            scanf("%d",&a);
            vis[a]=true;
            for(int i=1;i<=n;i++)
            {
                if(vis[i]&&dis[i][a]<=d)
                {
                    union_set(i,a);
                }
            }
        }
        else{
            scanf("%d%d",&a,&b);
            if(find_set(a)==find_set(b))
                printf("SUCCESS
");
            else
                printf("FAIL
");
        }
    }
    return 0;
}