POJ2236 Wireless Network ——并查集

​​​​​​POJ2236
题意：

有一个计算机网络的所有线路都坏了，网络中有n台计算机，现在你可以做两种操作，修理（O）和检测两台计算机是否连通（S），只有修理好的计算机才能连通。连通有个规则，两台计算机的距离不能超过给定的最大距离D（一开始会给你n台计算机的坐标）。检测的时候输出两台计算机是否能连通。

思路：

对于每次修理，遍历所有之前已经修好的机器，判断是否可以连通

注意输入输出格式

// Decline is inevitable,
// Romance will last forever.
// POJ2236
//#include <bits/stdc++.h>
#include <iostream>
#include <cmath>
#include <cstring>
#include <string>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <deque>
#include <vector>
using namespace std;
#define mst(a, x) memset(a, x, sizeof(a))
#define INF 0x3f3f3f3f
#define mp make_pair
#define pii pair<int,int>
#define fi first
#define se second
#define ll long long
//#define int long long
const int maxn = 1e3 + 10;
const int maxm = 25;
const int P = 1e9 + 7;
int fa[maxn];
int n, d;
int x[maxn], y[maxn];
int repair[maxn];
int find(int x) {
    if(x == fa[x]) return x;
    return fa[x] = find(fa[x]);
}
double dis(int a, int b) {
    return sqrt(double(x[a]-x[b]) * (x[a]-x[b]) + double(y[a]-y[b]) * (y[a]-y[b]));
}
void solve() {
    scanf("%d%d", &n, &d);
    for(int i = 0; i <= n; i++)
        fa[i] = i;
    for(int i = 0; i < n; i++) {
        cin >> x[i] >> y[i];
    }
    char cmd[2];
    int p,q,cnt=0;
    while(scanf("%s",cmd)!=EOF){
        if(cmd[0] == 'O')
        {
            scanf("%d", &p);
            p--;
            repair[cnt++] = p;
            for(int i = 0; i < cnt-1; i++)
                if(dis(repair[i], p) <= double(d) && repair[i] != p)
                    fa[find(p)] = find(repair[i]);
        }
        else
        {
//            cin >> p >> q;
            scanf("%d%d", &p, &q);
            p--, q--;
            if(find(p) == find(q))
//                puts("SUCCESS\n");
                printf("SUCCESS\n");
            else
//                puts("FAIL\n");
                printf("FAIL\n");
        }
        
    }
}
signed main() {
//    ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
//    int T; scanf("%d", &T); while(T--)
//    int T; cin >> T; while(T--)
    solve();
    return 0;
}