洛谷P1991 无线通讯网 最小生成树

题目链接：https://www.luogu.com.cn/problem/P1991

题目要求瓶颈生成树，最小生成树是瓶颈生成树的充分不必要条件。由于可以用卫星电话，所以我们可以这样想当我们把最小生成树求出来之后，将最长的那几条边用卫星电话连接，就可以使无线电范围尽可能小。如果有m个卫星电话，能连成m-1条边。所以我们在求最小生成树的时候，只需要求到n-m条即可。这里我用的kruskal算法。
代码如下

#include <bits/stdc++.h>
using namespace std;
const int maxn=505;
const int inf=0xfffffff;
struct node
{
    int from;
    int to;
    double cost;
    friend bool operator < (node a,node b)
    {
        return a.cost > b.cost;
    }
}p;
struct Nod
{
    int x,y;
}vet[maxn];
int fa[maxn];
int n,m;
priority_queue<node>q;//优先队列
int find(int x)
{
    if(x==fa[x])
    return x;
    return fa[x]=find(fa[x]);
}
double kruskal()//求生成树
{
    int cnt=0;
    double ans;
    while(!q.empty()&&cnt<n-m)
    {
        node temp=q.top();
        q.pop();
        int x=temp.from,y=temp.to;
        double c=temp.cost;
        int xx=find(x),yy=find(y);
        //cout<<x<<" "<<y<<endl;
        //cout<<fa[x]<<" "<<fa[y]<<endl;
        if(xx!=yy)
        ans=c,fa[xx]=yy,cnt++;//加入生成树,更新最大值
        //cout<<ans<<endl;
    }
    return ans;
}
double dist(int x1,int y1,int x2,int y2)
{
    return sqrt((x1-x2)*(x1-x2)*1.0+(y1-y2)*(y1-y2)*1.0);
}
int main()
{
    scanf("%d %d",&m,&n);
    for(int i=1;i<=n;i++)
    fa[i]=i;//初始祖先节点是自己
    for(int i=1;i<=n;i++)
    scanf("%d %d",&vet[i].x,&vet[i].y);
    for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++)
    {
        if(i!=j)
        p.from=i,p.to=j,p.cost=dist(vet[i].x,vet[i].y,vet[j].x,vet[j].y),q.push(p);
    }
    printf("%.2lf",kruskal());
    return 0;
}