POJ-2236（并查集）

Posted 2022-08-21 garrettwale

Wireless NetWork

POJ-2236

• 需要注意这里的树的深度需要初始化为0。
• 而且，find函数需要使用路径压缩，这里的unint合并函数也使用了优化（用一开始简单的合并过不了）。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<string>
#include<cmath>
using namespace std;
int n,d;
bool repaired[1002];
struct node
    int x;
    int y;
;
node com[1002];
int set[1002];
int rank1[1002];//rank1[i]表示i的深度
int find(int x)
    if(x==set[x])
        return set[x];
    return set[x]=find(set[x]);

void unint(int a,int b)
    int ta=find(a);
    int tb=find(b);
    if(ta!=tb)
        if(rank1[ta]<rank1[tb])
            set[ta]=tb;
        else
            set[tb]=ta;
            if(rank1[ta]==rank1[tb])
                rank1[ta]++;
            
        
    else return;

double compute(int a,int b)
    return (com[a].x-com[b].x)*(com[a].x-com[b].x)+(com[a].y-com[b].y)*(com[a].y-com[b].y);

int main()
    for(int i=0;i<1002;i++)
        set[i]=i;
        rank1[i]=0;
    
    cin>>n>>d;
    int x,y;
    for(int i=1;i<=n;i++)
        scanf("%d%d",&com[i].x,&com[i].y);
    
    char c;
    while(scanf("%c",&c)!=EOF)
        if(c=='O')//修复
            int num;
            scanf("%d",&num);
            repaired[num]=true;
            for(int i=1;i<=n;i++)
                if(i!=num&&repaired[i]&&compute(num,i)<=d*d)
                    unint(i,num);
                
            
        else if(c=='S')
            int x,y;
            scanf("%d%d",&x,&y);
            if(repaired[x]&&repaired[y])
                int t1=find(x);
                int t2=find(y);
                if(t1==t2)
                    cout<<"SUCCESS"<<endl;
                else cout<<"FAIL"<<endl;
            else
                cout<<"FAIL"<<endl;
            
        
    
    return 0;