【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 = ;

 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 =  ; i <= n; i++)
         f[i] = i,vis[i] = ;
 }

 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 = ; i < n ;i++){
         cin>>dx[i]>>dy[i];
     }

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