码迷,mamicode.com
首页 > 编程语言 > 详细

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

时间:2019-09-06 13:12:05      阅读:112      评论:0      收藏:0      [点我收藏+]

标签:还需   pre   vector   https   cst   str   amp   return   通过   

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;
}

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

标签:还需   pre   vector   https   cst   str   amp   return   通过   

原文地址:https://www.cnblogs.com/GarrettWale/p/11474031.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!