POJ 2236 Wireless Network

传送门：http://poj.org/problem?id=2236

解题思路：

利用了并查集

在操作一：

主要判断这个维修的电脑，和已经维修好的电脑的关系，如果他们的距离不大于题中限制的距离，就把他们和并在一起。如果大于就不做处理

在操做二：

就是判断这两个电脑是否在同一个集合中，如果是在同一个集合中，那么输出“SUCCESS",否则输出”FAIL“

实现代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <cstdlib>
using namespace std;

const int maxn=10010;

struct Point{
    int x,y;
}p[maxn];

int f[maxn];        //这是并查集
double w[maxn][maxn];  //这是用来记录第i个电脑，和其它所有电脑的关系。
bool r[maxn];             //这是记录电脑是否被维修好。


double dist(double x1,double y1,double x2,double y2){
    return sqrt((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2));
}

void init(int N){
    for(int i=1;i<=N;i++)
    for(int j=1;j<=N;j++){
        w[i][j]=dist(p[i].x,p[i].y,p[j].x,p[j].y);
    }

    for(int i=1;i<=N;i++){
        r[i]=false;
        f[i]=i;
    }
}

int findfa(int a){
    if(f[a]!=a)
        f[a]=findfa(f[a]);
    return f[a];
}

void unit(int a,int b){
    int fa=findfa(f[a]);
    int fb=findfa(f[b]);

    if(fa!=fb){
        f[fa]=fb;
    }
}
int main(){
    int N,D;
    scanf("%d%d",&N,&D);
    for(int i=1;i<=N;i++){
        scanf("%d%d",&p[i].x,&p[i].y);
    }

    init(N);
    char str[3];

    while(scanf("%s",str)!=EOF){
        if(str[0]=='S'){
            int a,b;
            scanf("%d%d",&a,&b);
            if(findfa(a)!=findfa(b)){
                printf("FAIL
");
            }
            else{
                printf("SUCCESS
");
            }

        }else{
            int a;
            scanf("%d",&a);
            r[a]=true;

            for(int i=1;i<=N;i++){
                if(w[a][i]<=D&&r[i]){
                    unit(a,i);
                }
            }
        }

    }

}

 代码2：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <vector>
using namespace std;

const int MAXN=1010;
int fa[MAXN];

struct Node{
    int x,y;
}node[MAXN];

double dist(Node a, Node b){
    double x=a.x-b.x;
    double y=a.y-b.y;
    return sqrt(x*x+y*y);
}

int findfa(int v){
    if(fa[v]!=v)
        fa[v]=findfa(fa[v]);
    return fa[v];
}

void unite(int u,int v){
    int fu=findfa(fa[u]);
    int fv=findfa(fa[v]);
    if(fu!=fv)
        fa[fu]=fv;
    return;
}

void init(int n){
    for(int i=1;i<=n;i++)
        fa[i]=i;
}

int main(){
    int N,d;
    scanf("%d%d",&N,&d);
    for(int i=1;i<=N;i++)
        scanf("%d%d",&node[i].x,&node[i].y);
    char str[10];
    vector<int> vec;
    init(N);
    while(scanf("%s",str)!=EOF){
        if(str[0]=='O'){
            int tmp;
            scanf("%d",&tmp);
            for(int i=0;i<vec.size();i++)
                if(dist(node[tmp],node[vec[i]])<=d)
                    unite(tmp,vec[i]);
            vec.push_back(tmp);
        }else{
            int u,v;
            scanf("%d%d",&u,&v);
            if(findfa(u)!=findfa(v))
                printf("FAIL
");
            else
                printf("SUCCESS
");
        }
    }

}