Day2 1522. 无线网络

Description

有一个由n台计算机组成的无线网络。(n <= 1001),正常情况下，每台计算机都能跟与它距离不超过d的任何计算机通讯(d <= 20000)。地震发生了。所有的计算机都陷入瘫痪。专家们试着一台一台地修复计算机，以恢复整个无线网络。有时在修复的过程中，他们需要测试一下某两台计算机能否通讯（如果他们能通过别的正常的计算机进行通讯，也算他们之间可以通讯，即“能否通讯”可以是间接的）。
你的任务，就是模拟修复网络的过程，并回答“能否通讯”的询问。

Input

第一行两个整数，N和d，N表示计算机的数目，d表示两台计算机直接可直接通讯的最大距离。接下来的N行，每行两个整数Xi,Yi，表示每台计算机的座标。接下来有许多行，每行都是一个操作（或者是修复操作，或者是询问操作）。
操作的格式如下：
O p (1 <= p <= N) 修复操作，表示修复编号为p的电脑；
S p q (1 <= p, q <= N) 询问操作，询问编号为p和编号为q的电脑能否通讯。
如果一台电脑尚未被修复，则它不能和任何电脑通讯。

Output

对于每个询问操作：如果能够通讯，输出一行SUCCESS；如果无法通讯，输出一行FAIL

Sample Input

4 1
0 1
0 2
0 3
0 4
O 1
O 2
O 4
S 1 4
O 3
S 1 4

Sample Output

FAIL
SUCCESS

Data Constraint

Hint

【限制】
对于50%的数据，N <= 300, 操作次数 <= 10000；
对于100%的数据，N <= 1001, 操作次数 <= 300000。

做法：暴力连边（距离ok就连），然后做并查集就好了
代码如下：

#include <cstdio>
#include <iostream>
#include <cstring>
#include <string>
#define rep(i, a, b)    for(int i = a; i <= b; i++)
#define eg(i, x)    for (int i = ls[x]; i; i = e[i].n)
#define N 1007
#define ll long long
using namespace std;
struct arr
{
    int t, n;
}e[N * 1000];

bool b[N];
ll dis, x[N], y[N];
int n, f[N], ls[N * 1000], tot;

ll pf(ll a) { return a * a; }

int find(int x)
{
    if (f[x] == 0) return x;
    f[x] = find(f[x]);
    return f[x];
}

ll read()
{
    ll s = 0;
    char ch = getchar();
    while (ch < '0' || ch > '9')    ch = getchar();
    while (ch >= '0' && ch <= '9')  s = s * 10 + ch - '0', ch = getchar();
    return s;
}

bool getdis(int q, int p)
{
    if (pf(x[q] - x[p]) + pf(y[q] - y[p]) <= dis)   return 1;
    return 0;
}

void add(int x, int y)
{
    e[++tot].t = y;
    e[tot].n = ls[x];
    ls[x] = tot;
}

int main()
{
    scanf("%d", &n);
    cin >> dis;
    dis = pf(dis);
    rep(i, 1, n)
        cin >> x[i] >> y[i];
    rep(i, 1, n - 1)
        rep(j, i + 1, n)
            if (getdis(i, j))
            {
                add(i, j);
                add(j, i);
            }
    char ch = getchar();
    int q, p;
    while (~scanf("%c", &ch))
    {
        q = read();
        if (ch == 'S')  
        {
            p = read();
            int q1 = find(p), p1 = find(q);
            if (q1 != p1 || !b[q] || ! b[p])    printf("FAIL\n");
            else printf("SUCCESS\n");
        }
        if (ch == 'O')
        {
            b[q] = 1;
            eg(i, q)
                if (b[e[i].t])
                {
                    int q1 = find(e[i].t), p1 = find(q);
                    if (q1 != p1)   f[p1] = q1;
                }
        }
    }
}