POJ——2236 Wireless Network（并查集模板题）

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

测试样例

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

题意： 有台处于平面的电脑都坏了。已知两台电脑的直接通信距离为（即若两台已经修复好的电脑它们的直接距离小于等于，则可以直接通信。）。现在进行一系列的操作：若为 ，则此操作是对编号为的电脑进行修复。若为，则是测试编号为和编号为的电脑能否通信，需要你输出答案。

解题思路： 一道很有趣的并查集，为什么呢？因为合并操作是要我们自己去进行的。系统每进行一次操作，我们都要对所有的计算机进行判断它们是否能通信。是否能通信也就是是否处于同一集合，这里即是模板并查集。重要的一点就是在于我们怎么去合并？我们要引入一个数组来判断电脑是否已经被修复，若没有被修复自然是无法合并，我们还要判断两台电脑之间的距离是否小于等于。这两个限制条件即决定了能否通信。 知道了这个，则此题易解。具体看代码。

AC代码

/*

*
*/
#include<iostream>  //POJ不支持

#define rep(i,a,n) for (int i=a;i<=n;i++)//i为循环变量，a为初始值，n为界限值，递增
#define per(i,a,n) for (int i=a;i>=n;i--)//i为循环变量， a为初始值，n为界限值，递减。
#define pb push_back
#define IOS ios::sync_with_stdio(false);cin.tie(0); cout.tie(0)
#define fi first
#define se second
#define mp make_pair

using namespace std;

const int inf = 0x3f3f3f3f;//无穷大
const int maxn = 1e3+4;//最大值。
typedef long long ll;
typedef long double ld;
typedef pair<ll, ll>  pll;
typedef pair<int, int> pii;
//*******************************分割线，以上为自定义代码模板***************************************//

int n,d;//每台电脑所能通信的最大距离。
bool vis[maxn];//判断电脑是否修好。
int point[maxn][2];//存储每台电脑的坐标
int father[maxn];//存储电脑所在集合。
int Find(int x){
  int r=x;
  while(r!=father[r]){
    r=father[r];
  }
  int i=x,j;
  while(father[i]!=r){
    j=father[i];
    father[i]=r;
    i=j;
  }
  return r;
}
int main(){
  //freopen("in.txt", "r", stdin);//提交的时候要注释掉
  IOS;
  while(cin>>n>>d){
    rep(i,1,n){
      cin>>point[i][0]>>point[i][1];
      father[i]=i;
      vis[i]=false;
    }
    string op;
    int temp1,temp2;
    while(cin>>op){
      if(op=="O"){
        cin>>temp1;
        vis[temp1]=true;
        rep(i,1,n){
          if(i!=temp1&&vis[i]){
            int dx=point[temp1][0]-point[i][0];
            int dy=point[temp1][1]-point[i][1];
            if(dx*dx+dy*dy<=d*d){
              int fx=Find(i);
              int fy=Find(temp1);
              if(fx!=fy){
                father[fx]=fy;
              }
            }
          }
        }
      }
      else{
        cin>>temp1>>temp2;
        if(Find(temp1)!=Find(temp2)){
          cout<<"FAIL"<<endl;
        }
        else{
          cout<<"SUCCESS"<<endl;
        }
      }
    }
  }
  return 0;
}