POJ 2236 Wireless Network [并查集] 《挑战程序设计竞赛》2.4

题目链接： POJ 2236 Wireless Network
题目大意：

有N台的计算机，要用他们组成一个无线网络。 两台计算机之间的距离小于D则可以直接相连，大于D，则需要有中间节点来间接连接这两台计算机。给出所有计算机的坐标，初始状态所有的计算机都是坏的， 需要进行修复。
工程师可以进行两种操作，修复一台计算机或者测试两台计算机的连通性
输入”O x”表示修复第i台计算机
输入”S x y”表示测试计算机x, y的连通性， 连通则输出”SUCCESS”, 不连通则输出”FAIL”

输入格式：

第一行 N, D
接下来N行， 每行两个数， 表示第i台计算机的坐标
接下来一直到输入结束，每行一个操作， 或者表示修复，或者表示测试

输出格式：

对于每个测试操作，连通则输出”SUCCESS”, 不连通则输出”FAIL”

题解：

典型的并查集操作。由于距离的限制， 每次修复一台计算机，只需要检查可以和它直接相连的计算机是否有已经被修复过的， 然后Union这两个集合即可。基于这个想法， 我在进行修复操作之前，先给这些电脑建了个图。为此付出了$n^2$ 的代价。对于这题来说，或许有些多此一举了，说不定反而浪费了时间。毕竟N最大只有1001。完全可以直接枚举所有的其他计算机。但我觉得当n再增大个十几倍或许建图就十分必要了。

另外， c的字符输入要谨慎。不小心就会超时。EOF还是要自己判断一下的。

代码：

#include <iostream>
#include <vector>
#include <cstring>
#include <cstdio>
#define MAXN 1010
using namespace std;

pair<int, int> computer[MAXN]; // 计算机的坐标
int N, D;
int f[MAXN]; // 并查集的父节点
bool repaired[MAXN]; // 是否修复过
vector<int> *adj; //图的邻接表表示

//是否相邻
bool is_adj(int a, int b) {
    int dx = computer[a].first - computer[b].first;
    int dy = computer[a].second - computer[b].second;
    return dx*dx + dy*dy <= D*D;
}

//建图
void buildGraph() {
    adj = new vector<int>[N+1];
    for (int i = 1; i <= N-1; i++) {
        for (int j = i+1; j <= N; j++) {
            if (is_adj(i, j)) {
                adj[i].push_back(j);
                adj[j].push_back(i);
            }
        }
    }
}

int Find(int x) {
    if (f[x] == x) return x;
    else return f[x] = Find(f[x]);
}

void Union(int x, int y) {
    int fx = Find(x);
    int fy = Find(y);
    if (fx != fy) {
        f[fx] = fy;
    }
}

int main() {
    //输入
    scanf("%d%d", &N, &D);
    for (int i = 1; i <= N; i++)
        scanf("%d%d", &computer[i].first, &computer[i].second);
    //初始化
    memset(repaired, false, sizeof(repaired));
    for (int i = 1; i <= N; i++) f[i] = i;
    buildGraph(); //建图

    //操作
    int operation;
    int x, y;
    while ((operation = getchar()) != EOF) { // AC
    //对比其他输入方式 
    //while (scanf("%c", &operation)) { // Time Limit Exceeded
    //while (cin >> operation) { // AC
        if (operation)
        if (operation == 'O') {
            scanf("%d", &x);
            repaired[x] = true;
            vector<int>::iterator i;
            for (i = adj[x].begin(); i != adj[x].end(); i++) {
                if (repaired[*i]) Union(x, *i);
            }
        } 
        else if (operation == 'S') {
            scanf("%d%d", &x, &y);
            if (Find(x) == Find(y)) printf("SUCCESS\n");
            else printf("FAIL\n");
        }
    }
    return 0;
}