并查集练习题（21.7.22）

Wireless Network
题意：告诉你计算机数n和两台计算机可以直接通信的最大距离d，然后告诉你这n台计算机的坐标，只有修过的计算机且两台计算机的距离不超过d的计算机可以通信，给你修过的计算机标号问你任意给定的两台计算机是否可以通信。
思路：按照并查集的做法，如果两台计算机可以通信，就将它们合并到一个节点。这里有限制条件是距离和已修，所以先判断是否已修，把修过的计算机标号用一个数组记录，标为1，然后判断距离，距离我想着一开始就把所有可能的情况都判断了，需要用到二重循环，看题目给的范围也不小，就先试一下，超时再另谋出路。挺好的，方法可行。所以先把所以情况都判断了，然后用一个二维数组记录两个计算机有通信的可能，标为1。后面的步骤就是并查集的常规操作，在两个限制条件都满足的情况下把计算机合并。注意这里，告诉你修一台计算机，就要把它和其他已修过的且可以通信的计算机合并。也就是修一个，合并一堆，要把能和它合并的都合并。最后再判断两台计算机是否可以通信也就是是否合并到一个根结点上。

#include <iostream>
using namespace std;
const int maxn=2e4+5;
int s[maxn],x[maxn],y[maxn],z[maxn];
int n,d,h[maxn][maxn],p,m,q;
char ch;
void init_set(int a)
{
    for(int i=0; i<=a; i++)
    {
        s[i]=i;
    }
}
int find_set(int a)
{
    return s[a]==a?a:s[a]=find_set(s[a]);
}
void union_set(int a,int b)
{
    a=find_set(a);
    b=find_set(b);
    if(a!=b)
        s[a]=b;
}
int main()
{
    cin>>n>>d;
    init_set(n);
    for(int i=1; i<=n; i++)
    {
        cin>>x[i]>>y[i];
    }
    for(int i=1; i<=n; i++)
    {
        for(int j=i; j<=n; j++)
        {
            p=(x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]);
            if(p<=d*d)
            {
                h[i][j]=1;
                h[j][i]=1;
            }
        }
    }
    while(cin>>ch)
    {
        if(ch=='O')
        {
            cin>>m;
            z[m]=1;
            for(int i=1; i<=n; i++)
            {
                if(h[m][i]&&z[i])
                {
                    union_set(m,i);
                }
            }
        }
        if(ch=='S')
        {
            cin>>m>>q;
            if(find_set(m)==find_set(q))
            {
                cout<<"SUCCESS"<<endl;
            }
            else cout<<"FAIL"<<endl;
        }
    }
    return 0;
}

A Bug’s Life
题意：告诉你虫子的数量和交互组数，让你判断是否交互错误。（就是给你让你分析是否给出的数据中存在相同的性别）
分析：这个题的大意抽象出来就是维护不同集合，也就是告诉你a，b在不同派，不懂点这里
一开始题没读懂，没想性别的事，觉得只要是有两个以上的合并在一起就不对，就比如1 2，2 3，1 3，最后肯定1 2 3都合并在一起了，合并了3个，肯定不对。但这样没有考虑性别，同一个性别下合并的肯定不止2个，所以不对。
正确的做法就是用维护不同集合的方法，用unite_set(a,b)表示a,b属于一个性别，用unite_set(a,b+n)和unite_set(a+n,b)表示a,b属于不同性别，一开始交互的肯定都是不同性别，后来如果遇到同性别的那么a与a+n一定会合并，这样来判断是否交互错误。
这题还比较坑的地方就是用cin，cout会超时，另外输出格式也要注意。😑

#include <iostream>
#include <cstdio>
using namespace std;
const int maxn=5000;
int s[maxn+1],y[maxn+1];
void init_set()//初始化
{
    for(int i=0; i<=maxn; i++)
    {
        s[i]=i;
    }
}
int find_set(int x)//查询
{
    return x==s[x]?x:find_set(s[x]);
}
void union_set(int x,int y)//合并
{
    x=find_set(x);
    y=find_set(y);
    if(x!=y)s[x]=y;
}
int main()
{
    int t,n,m,a,b,d=1,p;
    scanf("%d",&t);
    while(t--)
    {
        init_set();
        p=0;
        scanf("%d%d",&n,&m);
        for(int i=1; i<=m; i++)
        {
            scanf("%d%d",&a,&b);
            union_set(a,b+n);
            union_set(a+n,b);
            if(find_set(a)==find_set(a+n)||find_set(b)==find_set(b+n))
            {
                p=1;
            }
        }
        if(d>1)
            printf("\n");
        printf("Scenario #%d:\n",d++);
        if(p)
        {
            printf("Suspicious bugs found!\n");
        }
        else
        {
            printf("No suspicious bugs found!\n");
        }
    }
    return 0;
}