公路修建 最小生成树prim算法

有$n$个城市,政府决定修建公路,在每一轮中，每个城市选择一个与它最近的城市，申请修建通往该城市的公路。政府负责审批这些申请以决定是否同意修建。政府审批的规则如下：

如果两个或以上城市申请修建同一条公路，则让它们共同修建；
如果三个或以上的城市申请修建的公路成环。政府将否决其中最短的一条公路申请；

计算出将要修建的公路总长度。
首先审题，可以知道城市选择一个与它最近的城市，再计算公路总长度，这可以看出最小生成树的板子。再看规则一说明是无向图，而规则二其实不起作用，是迷惑项。因为每个城市找最近的，所以三个以上城市成环的情况是不可能出现的。
最小生成树有两个算法一个是Kruskal，它适用于稀疏图。而本题属于稠密图，使用Kruskal会超时，所以改用Prim算法。

#include<bits/stdc++.h>
using namespace std;
double x[5010],y[5010],dis[5010];
bool flag[5010];
double minn,ans=0;
int n,now;
double distance(double x1,double x2,double y1,double y2)
{
	return sqrt((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2));
}
void prim()
{
	dis[1]=0;flag[1]=1;
	for(int i=1;i<=n;i++)
	{
		now=1;minn=999999.9;
		for(int j=1;j<=n;j++)
		if(flag[j]!=1&&dis[j]<minn)
		{
			minn=dis[j];now=j;
		}
		flag[now]=1;ans+=dis[now];
		for(int j=1;j<=n;j++)
		dis[j]=min(dis[j],distance(x[now],x[j],y[now],y[j]));
	}
}
int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>x[i]>>y[i];
		dis[i]=9999999.9;
	}
	prim();
	printf("%.2lf",ans);
	return 0;
}