POJ-2236-Wireless Network [并查集]

---

题目传送门

---

题意：
一个地图上分布着N台电脑，并给出N台电脑的坐标，如果两者之间距离小于等于d就可以通信，可以间接通信。
给出操作”O x”为修好x， 操作”S x y”为询问x y是否可以通信。输出”FAIL”或”SUCCESS”
思路：
如果输入”O x”就枚举所有已经出现过的点，判断两点之间距离是否小于等于d，如果小于等于d，就将这两个点合并。
如果输入”S x y”就求x y的父节点，相同即成功，不同即失败。

#include <algorithm>#include <iostream>#include <cstdlib>#include <cstring>#include <cstdio>#include <cmath>using namespace std;struct node{    int x,y;}Point[1200];int vis[1200];int Father[1200];int n,d;void init(){    for (int i = 0; i < 1200; i++)    {        Father[i] = i;    }    memset(vis,0,sizeof(vis));}int Find(int x){    while (Father[x]!=x)        x = Father[x];    return x;}bool Dis(int a, int b){    int p = Point[a].x - Point[b].x;    int q = Point[a].y - Point[b].y;    if (p*p+q*q<=d*d)        return true;    return false;}int main(void){    //freopen("in.txt","r",stdin);    //freopen("out.txt","w",stdout);    scanf("%d %d", &n, &d);    init();    for (int i = 1; i <= n; i++)        scanf("%d %d", &Point[i].x, &Point[i].y);    char ch;    while (cin>>ch)    {        if (ch=='O')        {            int r;            scanf("%d", &r);            vis[r] = 1;            for (int i = 1; i <= n; i++)            {                if (i!=r && vis[i] && Dis(i,r))                {                    int p = Find(i);                    int q = Find(r);                    Father[p] = q;                }            }        }        else        {            int r1,r2;            scanf("%d %d", &r1, &r2);            if (Find(r1)==Find(r2))                printf("SUCCESS\n");            else                printf("FAIL\n");        }    }    return 0;}