【POJ - 2236】Wireless Network （并查集）

佚名 5年前 (2019-08-10) 算法 1564人围观抢沙发百度已收录

Wireless Network

这接翻译了

Descriptions

SRE实战互联网时代守护先锋，助力企业售后服务体系运筹帷幄！一键直达领取阿里云限量特价优惠。地震发生在东南亚。ACM（亚洲合作医疗团队）已经与膝上电脑建立了无线网络，但是一次意外的余震袭击，网络中的所有计算机都被打破了。计算机一个接一个地修复，网络逐渐开始工作。由于硬件限制，每台计算机只能直接与距离它不远的计算机进行通信。但是，每台计算机都可以被视为两台其他计算机之间通信的中介，也就是说，如果计算机A和计算机B可以直接通信，或者计算机C可以与A和A进行通信，则计算机A和计算机B可以进行通信。 B.

在修复网络的过程中，工作人员可以随时进行两种操作，修复计算机或测试两台计算机是否可以通信。你的工作是回答所有的测试操作。

Input

第一行包含两个整数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行。

Output

对于每个测试操作，如果两台计算机可以通信则打印“SUCCESS”，否则打印“FAIL”。

Sample Output

Sample Output

FAIL
SUCCESS

题目链接

https://vjudge.net/problem/POJ-2236

每次修理好一台计算机的时候就遍历一下所有修好的计算机，看距离是否<=d，如果符合说明可以连通，将两台计算机所在集合合并。 每次检查的时候判断一下这两台计算机是否在同一集合中即可。 坐标之后给出的计算机编号都是n+1的。例如O 3，他实际上修理的是编号为2的计算机，因为计算机是从0开始编号的 不会并查集的可以看看这个 https://www.cnblogs.com/sky-stars/p/11222999.html AC代码

#include <iostream>
#include <cstdio>
#include <fstream>
#include <algorithm>
#include <cmath>
#include <deque>
#include <vector>
#include <queue>
#include <string>
#include <cstring>
#include <map>
#include <stack>
#include <set>
#include <sstream>
#define IOS ios_base::sync_with_stdio(0); cin.tie(0);
#define Mod 1000000007
#define eps 1e-6
#define ll long long
#define INF 0x3f3f3f3f
#define MEM(x,y) memset(x,y,sizeof(x))
#define Maxn 100000+5
using namespace std;
int n,d;
int dx[Maxn];//坐标
int dy[Maxn];
int par[Maxn];//par[i] i的根
int pairs[Maxn];//pairs[i]=p 第i个修好的电脑编号为p
int findr(int x)//查询根
{
    if(par[x]==x)
        return x;
    return par[x]=findr(par[x]);
}
void unite(int x,int y)//合并
{
    x=findr(x);
    y=findr(y);
    if(x==y)//根相同不用管
        return;
    par[x]=y;//若根不同，y并入x，则y的根为x，同理x也能并入y  这里随意
}
double dis(int a,int b)//求距离
{
    return sqrt( (double)((dx[a]-dx[b])*(dx[a]-dx[b]) + (dy[a]-dy[b])*(dy[a]-dy[b])) );
}
int main()
{
    MEM(pairs,0);
    cin>>n>>d;
    for(int i=0; i<n; i++)
        cin>>dx[i]>>dy[i];//集合初始化
    for(int i=0; i<n; i++)
        par[i]=i;
    int p,q,len=0;
    char op;
    while(cin>>op)
    {
        if(op=='O')
        {
            cin>>p;
            p--;
            pairs[len++]=p;
            for(int i=0; i<len; i++)
            {
                if(dis(pairs[i],p)<=(double)d)//判断距离
                    unite(pairs[i],p);//合并集合
            }
        }
        else
        {
            cin>>p>>q;
            p--,q--;
            if(findr(p)==findr(q))//查根
                cout<<"SUCCESS"<<endl;
            else
                cout<<"FAIL"<<endl;
        }
    }
    return 0;
}