(并查集模板题)Wireless Network POJ - 2236

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

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

输入

第一行包含了两个整数 N 和 d (1 <= N <= 1001, 0 <= d <= 20000)。此处 N 是电脑的数目，编号从 1 到 N；同时，D 是两台电脑之间能够直接通信的最大距离。接下来的 N 行，每行包含两个整数 xi, yi (0 <= xi, yi <= 10000)，表示 N 台电脑的坐标。从第 (N+1) 行到输入结束，是逐一执行的操作，每行包含一个操作，格式是以下两者之一：
1. "O p" (1 <= p <= N)，表示维护电脑 p 。
2. "S p q" (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

思路：

对于O操作，每修理一个电脑就检测该电脑与所有被检测过的电脑之间能否通信，可以就把两个电脑加入同一个集合中
对于S操作，判断两台电脑是否在都修理过且在同一个集合中，满足就输出SUCCESS，否则输出FAIL

代码： 

#include <iostream>
#include <algorithm>
#include <string.h>
#include <cstdio>
#include <string>
#include <cmath>
#include <vector>
#include <stack>
#include <queue>
#include <stack>
#include <list>
#include <map>
#include <set>
//#include <unordered_map>
#define Fbo friend bool operator < (node a, node b)
#define mem(a, b) memset(a, b, sizeof(a))
#define FOR(a, b, c) for (int a = b; a <= c; a++)
#define RFOR(a, b, c) for (int a = b; a >= c; a--)
#define off ios::sync_with_stdio(0)
#define sc(a) scanf("%d",&a) 
bool check1(int a) { return (a & (a - 1)) == 0 ? true : false; }

using namespace std;
typedef pair<int, int> pii;
typedef long long ll;
const int INF = 0x3f3f3f3f;
const int mod = 1e9 + 7;
const int Maxn = 1e5 + 5;
const double pi = acos(-1.0);
const double eps = 1e-8;

int a[Maxn],n,d;
int b[Maxn], cnt=0;//被修理的电脑编号
bool vis[Maxn];

struct node {
    int x, y;
}point[Maxn];

void init() {
    for (int i = 1; i < Maxn; i++)
        a[i] = i;
}

int find(int x) {
    return x == a[x] ? x : find(a[x]);
}

void unite(int x, int y) {
    x = find(x);
    y = find(y);
    if (x != y) {
        a[x] = a[y];
    }
}

bool check(int a, int b) {
    double dis = sqrt((double)((point[a].x - point[b].x) * (point[a].x - point[b].x)) + ((point[a].y - point[b].y) * (point[a].y - point[b].y)));
    //cout << dis << "---" << d << endl;
    return dis <= d;
}

int main() {
    off;
    mem(vis, 0);
    cin >> n >> d;
    init();
    for (int i = 1; i <= n; i++) {
        cin >> point[i].x >> point[i].y;
    }
    char ch;
    while (cin >> ch) {
        if (ch == 'O') {
            int m;
            cin >> m;
            vis[m] = 1;
            for (int i = 0; i < cnt; i++) {
                if (check(b[i], m)) {
                    unite(b[i], m);
                }
            }
            b[cnt++] = m;
        }
        else {
            int l, r;
            cin >> l >> r;
            if (vis[l] && vis[r] && find(l) == find(r)) {
                cout << "SUCCESS" << endl;
            }
            else cout << "FAIL" << endl;
        }
    }
    return 0;
}