【POJ】2236 Wireless Network

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

 

题意：给你n台计算机的坐标。d是可通信的最大距离。有两个操作。

1、O p 表示修复计算机p.

2、S p q表示询问pq是否能够通信。

 

题解：并查集的提升。把距离考虑在判断内。如果修复了p就对当前集合做一个并操作。查找的时候直接判断父亲是不是共同的即可。

 

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <cmath>
#include <vector>
#include <stack>
#include <queue>
#define ll long long
using namespace std;
const int maxn = 1010;

int dx[maxn],dy[maxn];
int f[maxn],vis[maxn];
int n,d;

int dis(int a,int b){
    int distance = (dx[a]-dx[b]) * (dx[a] - dx[b]) + (dy[a]-dy[b]) * (dy[a] - dy[b]);
    if(distance <= d*d)    return true;
    else    return false;
}

void init(int n){
    for(int i = 0 ; i <= n; i++)
        f[i] = i,vis[i] = 0;
}

int find(int x){
    if(x == f[x])
        return x;
    return f[x] = find(f[x]); 
} 

void join(int a, int b){
    a = find(a);
    b = find(b);
    if(a != b){
        f[a] = b;
    }
}

int main(){
    
    cin>>n>>d;
    for(int i = 0; i < n ;i++){
        cin>>dx[i]>>dy[i];
    }

    init(n);
    char ch;
    int num;
    int cnt = 0;
    while(cin>>ch){
        if(ch == 'O'){
            cin>>num;
            num--;
            vis[cnt++] = num;
            for(int i = 0; i < cnt-1; i++){
                if(vis[i] != num && dis(vis[i],num)){
                    join(vis[i],num);
                }
            }
        }
        else{
            int p,q;
            cin>>p>>q;
            if(find(p-1) == find(q-1)){
                cout<<"SUCCESS"<<endl;
            }
            else{
                cout<<"FAIL"<<endl;
            }
        }
    }
    return 0;
}