POJ2236 Wireless Network(并查集)

描述

传送门：我是传送门

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

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

输入

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

1. “O p” (1<=p<=N)(1<=p<=N)，表示维护电脑 p 。 

2. “S p q” (1<=p,q<=N)(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

思路

简单的并查集问题，主要是注意处理距离比较问题

为避免因为计算距离的时候出现精度误差，直接d取平方，不进行开方
用use数组标记修过的网络，只有修过的才对其进行更新

代码

/*
 * =================================================================
 *
 *       Filename:  A.cpp
 *
 *           Link:  http://poj.org/problem?id=2236
 *
 *        Version:  1.0
 *        Created:  2018/09/18 10时15分26秒
 *       Revision:  none
 *       Compiler:  g++
 *
 *         Author:  杜宁元 (https://duny31030.top/), duny31030@126.com
 *   Organization:  QLU_浪在ACM
 *
 * =================================================================
 */
#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <math.h>
#include <string.h>
using namespace std;
#define clr(a, x) memset(a, x, sizeof(a))
#define rep(i,a,n) for(int i=a;i<=n;i++)
#define pre(i,a,n) for(int i=n;i>=a;i--)
#define ll long long
#define max3(a,b,c) fmax(a,fmax(b,c))
#define ios ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0);
const double eps = 1e-6;
const int INF = 0x3f3f3f3f;
const int mod = 1e9 + 7;
const int N = 1e4+10;
int f[N],x[N],y[N];
int find(int i) 
{  
    if(i!=f[i]) 
        f[i]=find(f[i]);  
    return f[i];  
}  
int dis(int xx,int yy)
{
    int x1 = x[xx]-x[yy];
    x1 *= x1;
    int x2 = y[xx]-y[yy];
    x2 *= x2;
    return (x1+x2);
}
int n,d,cx,cy;
int use[N];
int main()
{
    ios
#ifdef ONLINE_JUDGE 
#else 
        freopen("in.txt","r",stdin);
    // freopen("out.txt","w",stdout); 
#endif
    // scanf("%d %d",&n,&d);
    cin >> n >> d;
    clr(use,0);
    int t = d*d;
    rep(i,1,n)
    {
        f[i] = i;
        // scanf("%d %d",&x[i],&y[i]);
        cin >> x[i] >> y[i];
    }
    char op;
    while(cin >> op)
    {
        if(op == 'O')
        {
            // scanf("%d",&cx);
            cin >> cx;

            use[cx] = 1;
            rep(i,1,n)
            {
                if(cx != i && use[i] && dis(i,cx) <= t)
                {
                    int k1 = find(cx);
                    int k2 = find(i);
                    if(k1 < k2)
                        f[k1] = k2;
                    else 
                        f[k2] = k1;
                }
            }
        }
        else 
        {
            // scanf("%d %d",&cx,&cy);
            cin >> cx >> cy;
            int fx = find(cx);
            int fy = find(cy);
            if(fx == fy)
                cout << "SUCCESS" << endl;
            else 
                cout << "FAIL" << endl;
        }
    }
    fclose(stdin);
    // fclose(stdout);
    return 0;
}