Wireless Network 题解 并查集

题目链接: vjudge
芜湖! 很开心刚学了并查集就把题做出来了！

题意

南亚发生了一次地震。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

解题思路

首先要学会并且理解并查集的思想。每次修好一个电脑，我们就把它和其他所有修好的电脑进行一个合并操作。
易错点：

1. 距离使用浮点数表示
2. 不能和自己合并

AC代码

#include <stdio.h>
#include <vector>
#include <cstring>
#include <math.h>
#define Num 300005
#define Max 0x3f3f3f3f
using namespace std;
typedef struct
{
    int x, y;
} pos;

vector<int> pre;
vector<int> erank;
vector<pos> v(1);
int find(int a)
{
    if (pre[a] == a)
        return a;
    return pre[a] = find(pre[a]);
}
void join(int a, int b)
{
    int x = find(a);
    int y = find(b);

    if (erank[x] < erank[y])
    {
        pre[x] = y;
    }
    else
    {
        if (erank[x] == erank[y])
        {
            erank[x]++;
        }
        pre[y] = x;
    }
}
double getlen(pos x, pos y)
{
    return sqrt((x.x - y.x) * (x.x - y.x) + (x.y - y.y) * (x.y - y.y));
}
int main()
{
    int n, d;

    scanf("%d %d", &n, &d);
    bool vis[n + 1];
    memset(vis, false, sizeof vis);
    pre.resize(n + 1);
    erank.resize(n + 1);
    for (int i = 1; i <= n; i++)
    {
        pre[i] = i;
        erank[i] = 1;
    }

    for (int i = 0; i < n; i++)
    {
        pos buf;
        scanf("%d %d", &buf.x, &buf.y);
        v.push_back(buf);
    }
    char c;
    int p, q;
    while (scanf("%c", &c) != EOF)
    {
        if (c == 'O')
        {
            scanf("%d", &p);
            vis[p] = true;
            for (int i = 1; i <= n; i++)
            {
                if (vis[i] == true && getlen(v[i], v[p]) <= d && i != p)
                {
                    join(i, p);
                }
            }
        }
        else if (c == 'S')
        {
            scanf("%d %d", &p, &q);
            if (find(p) == find(q))
            {
                printf("SUCCESS\n");
            }
            else
            {
                printf("FAIL\n");
            }
        }
        // for (int i = 1; i <= n; i++)
        // {
        //     printf("%d ", pre[i]);
        // }
        // printf("\n");
        // for (int i = 1; i <= n; i++)
        // {
        //     printf("%d ", erank[i]);
        // }
        // printf("\n");
    }
    return 0;
}