POJ-2349(kruskal算法+最小生成树中最大边的长度)

Arctic

POJ-2349

• 这题是最小生成树的变形题目。题目的意思是已经有s个卫星频道，这几个卫星频道可以构成一部分的网络，而且不用费用，剩下的需要靠d的卫星接收器。题目要求的就是最小生成树中，最大的边的长度。
• 题目中的传入kruskal函数里面的sn表示还需要连接的顶点个数，因为剩下的可以使用卫星频道来连接。
• 剩下的就是kruskal的问题了，这里通过当前最大边来返回答案。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#include<vector>
#include<cmath>
using namespace std;
const int maxn=505;
const int maxm=300000;
const double INF=0X3F3F3F3F;
int s,n;
int e;
struct node{
    int from;
    int to;
    double cost;
    bool operator<(const node& t)const{
        return cost<t.cost;
    }
};
node edge[maxm];
int x[maxn];
int y[maxn];
int set[maxn];
int find(int x){
    return set[x]==x?set[x]:set[x]=find(set[x]);
}
double kruskal(int sn){
    sort(edge,edge+e);
    for(int i=1;i<=n;i++){
        set[i]=i;
    }
    double maxs=-INF;
    int ans=0;
    for(int i=0;i<e;i++){
        int from=edge[i].from;
        int to=edge[i].to;
        double cost=edge[i].cost;
        int x=find(from);
        int y=find(to);
        if(x!=y){
            set[x]=y;
            maxs=max(cost,maxs);
            ans++;
        }
        if(ans==sn-1){
            break;
        }
    }
    if(ans<sn-1){
        return -1;
    }else{
        return maxs;
    }
}
int main(){
    int t;
    scanf("%d",&t);
    while(t--){
        e=0;
        scanf("%d%d",&s,&n);
        for(int i=1;i<=n;i++){
            scanf("%d%d",&x[i],&y[i]);
        }
        for(int i=1;i<=n;i++){
            for(int j=1;j<i;j++){
                double cost=sqrt((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]));
                edge[e++]=node{i,j,cost};
            }
        }
        printf("%.2f\n",kruskal(n-s+1));
    }
    return 0;
}