poj 2236 Wireless Network 并查集

题目

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

题目来源：《挑战》练习题

简要题意：$N$个节点坏了，距离小于等于$d$，而且俩点修好了它们之间就能通信。
$\qquad \quad \ \ \ \ $给定询问，修好一个节点或者问两点之间是否能通信，可以间接通信。

数据范围：$1 \leqslant N \leqslant 1001;\quad 0 \leqslant d \leqslant 20000;\quad 0 \leqslant x_i, y_i \leqslant 10000$

题解

赤裸裸的并查集咯，每次修好一个点，对于距离满足且修好的点把它们合并。

查询就是直接看是否属于一个祖先。

代码

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <stack>
#include <queue>
#include <string>
#include <vector>
#include <set>
#include <map>

#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define sz(x) ((int)(x).size())
#define fi first
#define se second
using namespace std;
typedef long long LL;
typedef vector<int> VI;
typedef pair<int,int> PII;
LL powmod(LL a,LL b, LL MOD) {LL res=1;a%=MOD;for(;b;b>>=1){if(b&1)res=res*a%MOD;a=a*a%MOD;}return res;}
// head
const int N = 1000+5;
int r[N];
int fa[N];

int find(int key) {
    return (key == fa[key]) ? key : (fa[key] = find(fa[key]));
}

void init(int n) {
    for (int i = 0; i <= n; i++) {
        fa[i] = i;
    }
}

void joint(int u, int v) {
    int a = find(u), b = find(v);
    if (a != b) {
        if (r[a] > r[b]) swap(a, b);
        fa[a] = b;
        if (r[a] == r[b]) r[b]++;
    }
}

bool same(int u, int v) {
    return find(u) == find(v);
}

inline double sq(double x) {
    return x*x;
}
inline double dis(const PII &a, const PII &b) {
    return sqrt(sq(a.fi-b.fi) + sq(a.se-b.se));
}

PII a[N];
bool vis[N];
char s[5];
int main()
{
    int n, d;
    scanf("%d%d", &n, &d);
    for (int i = 1; i <= n; i++) {
        scanf("%d%d", &a[i].fi, &a[i].se);
    }

    init(n);
    int u, v;
    while (scanf("%s", s) == 1) {
        if (s[0] == 'O') {
            scanf("%d", &u);
            if (vis[u]) continue;
            vis[u] = true;
            for (int i = 1; i <= n; i++) {
                if (vis[i] && dis(a[u], a[i]) <= d) {
                    joint(u, i);
                }
            }
        } else {
            scanf("%d%d", &u, &v);
            puts(same(u, v) ? "SUCCESS" : "FAIL");
        }
    }
    return 0;
}