洛谷P1265 公路修建 最小生成树

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

题目要求城市要与距离自己最近的城市修建公路，乍一看这不是最小生成树吗？！但本题还有一个条件，当三个城市要连成环时，要否决距离最短的那条公路。但是，除非是等腰三角形，不可能存在连成环的情况，但如果是等腰三角形。就可以连成等腰或者不连成等腰，这与题目唯一解就相矛盾了，所以还是求最小生成树。这里用prime，因为城市数量最多有5000，如果用kruskal，必须预处理每条边长度，会爆内存，但用prime时间是允许的，并且求城市之间的距离可以现用现求，不会爆内存。
代码如下

#include <bits/stdc++.h>
using namespace std;
const int maxn=5010;
const int inf=0xfffffff;
//double vertex[maxn][maxn];//邻接矩阵
bool vis[maxn];//判断是否已在生成树中
double lowcost[maxn];//以某点为终点的最短距离
int n,m;
struct node
{
    int x,y;
}p[maxn];
double dist(int x1,int y1,int x2,int y2)
{
    return sqrt((double)(x1-x2)*(x1-x2)+(double)(y1-y2)*(y1-y2));
}
double prime()//求最小生成树
{
    double ans=0.0;//记录生成树长度
    vis[1]=true;//以编号为1的点为起始点
    for(int i=1;i<=n;i++)
    lowcost[i]=inf;
    for(int i=1;i<=n;i++)
    if(!vis[i]&&lowcost[i]>dist(p[1].x,p[1].y,p[i].x,p[i].y))
    lowcost[i]=dist(p[1].x,p[1].y,p[i].x,p[i].y);
    for(int i=1;i<n;i++)
    {
        double Mi=inf;
        int t;
        for(int j=1;j<=n;j++)
        if(!vis[j]&&Mi>lowcost[j])//没在生成树中，找最短的
        Mi=lowcost[j],t=j;
        vis[t]=true;//将这个点加入生成树
        ans+=Mi;//更新生成树长度
        for(int j=1;j<=n;j++)
        {
            double temp=dist(p[t].x,p[t].y,p[j].x,p[j].y);
            if(!vis[j]&&lowcost[j]>temp)//更新以到各点的最小值
            lowcost[j]=temp;
        }
    }
    return ans;
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    scanf("%d %d",&p[i].x,&p[i].y);
    printf("%.2lf\n",prime());
    return 0;
}