Arctic Network——最小生成树

Posted 2022-12-12 j666

题目链接

题意：

给出p个节点的坐标，S个卫星，每个节点相互之间连通且边权为两点间距离，求这个图的最小生成树。

S个卫星相当于S-1条边权值为0的边。

题解：

输出最小生成树中除去最长的s-1条边后最长的边的边权（即 （p-1）-（s-1）），保留两位小数。

因为 求的最小权值所以 让最长的的s-1条边权值为0

所以求第 p-s 条边的边权

 

代码：

#include<iostream>
#include<stdio.h>
#include<math.h>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long ll;
const int maxn = 2010;
int f[maxn];
double x[maxn],y[maxn];
int n,cnt,p,s;
struct node

    int u,v;
    double w;
    bool operator < (const node &a)const
    
        return w<a.w;
    
 edge[maxn*maxn];

int Find(int x)

    return x==f[x]?x:f[x]=Find(f[x]);

void add(int u,int v,double w)

    edge[cnt].u=u;
    edge[cnt].v=v;
    edge[cnt++].w=w;


double kruskal()

    int sum=0;
    for(int i=0; i<=p; i++)f[i]=i;
    sort(edge,edge+cnt);
    for(int i=0; i<cnt; i++)
    
        int x=edge[i].u;
        int y=edge[i].v;
        int fx=Find(x);
        int fy=Find(y);
        if(fx!=fy)
        
            sum++;
            f[fx]=fy;
        
        if(sum==p-s)return edge[i].w;
    
    return 0.0;

double get_dis(int i,int j)

    double dis=sqrt((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]));
    //printf("%.2f\n",dis);
    return dis;

int main()

    scanf("%d",&n);
    while(n--)
    
        scanf("%d%d",&s,&p);

        for(int i=1;i<=p;i++)scanf("%lf%lf",&x[i],&y[i]);
        cnt=0;
        for(int i=1;i<=p;i++)for(int j=i+1;j<=p;j++)
        
            double dis=get_dis(i,j);
            add(i,j,dis);
        
        printf("%.2f\n",kruskal());
    
    return 0;

kruskal