公路修建 P1265

题目描述

某国有n个城市，它们互相之间没有公路相通，因此交通十分不便。为解决这一“行路难”的问题，政府决定修建公路。修建公路的任务由各城市共同完成。

修建工程分若干轮完成。在每一轮中，每个城市选择一个与它最近的城市，申请修建通往该城市的公路。政府负责审批这些申请以决定是否同意修建。

政府审批的规则如下：

（1）如果两个或以上城市申请修建同一条公路，则让它们共同修建；

（2）如果三个或以上的城市申请修建的公路成环。如下图，A申请修建公路AB，B申请修建公路BC，C申请修建公路CA。则政府将否决其中最短的一条公路的修建申请；

（3）其他情况的申请一律同意。

一轮修建结束后，可能会有若干城市可以通过公路直接或间接相连。这些可以互相：连通的城市即组成“城市联盟”。在下一轮修建中，每个“城市联盟”将被看作一个城市，发挥一个城市的作用。

当所有城市被组合成一个“城市联盟”时，修建工程也就完成了。

你的任务是根据城市的分布和前面讲到的规则，计算出将要修建的公路总长度。

输入输出格式

输入格式：

第一行一个整数n，表示城市的数量。(n≤5000)

以下n行，每行两个整数x和y，表示一个城市的坐标。(-1000000≤x，y≤1000000)

输出格式：

一个实数，四舍五入保留两位小数，表示公路总长。（保证有惟一解）

输入输出样例

输入样例#1：

4
0 0
1 2
-1 2
0 4

输出样例#1：

6.47

说明

修建的公路如图所示：

其实就是求最小生成树，注意规则（2）是不会有影响的，证明如下：

因为A申请修建公路AB，所以AB<=AC,B申请修建公路BC,所以BC<=AB,C申请修建公路CA,所以AC<=BC,而这三个条件同时成立当且仅当AB=BC=AC，既然这样，去掉那条边都是一样的，这与求解最小生成树不冲突。

因为这是一个完全图，边数较多，所以我用了堆优化的prime算法，速度非常快，600ms。

#include<iostream>
#include<cstring>
#include<cmath>
#include<iomanip>
#define f(i,l,r) for(i=(l);i<=(r);i++)
using namespace std;
const int MAXN=5005;
struct Node{
	int x,y;
}a[MAXN];
int n,vis[MAXN],num;
double d[MAXN],ans=0,heap0[10000000];
int heap1[10000000],sz;
inline double cal(int u,int v)
{
	return sqrt((double)(a[u].x-a[v].x)*(a[u].x-a[v].x)+(double)(a[u].y-a[v].y)*(a[u].y-a[v].y));
}
inline void pushup(int p)
{
	int fa=p>>1,id=heap1[p];
	double a=heap0[p];
	while(fa&&a<heap0[fa]){
		heap0[p]=heap0[fa];
		heap1[p]=heap1[fa];
		p=fa;
		fa>>=1;
	}
	heap0[p]=a;
	heap1[p]=id;
}
inline void pushdown(int p)
{
	int son=p<<1,id=heap1[p];
	double a=heap0[p];
	while(son<=sz){
		if(son<sz&&heap0[son]>heap0[son+1]) son++;
		if(heap0[son]>=a) break;
		heap0[p]=heap0[son];
		heap1[p]=heap1[son];
		p=son;
		son<<=1;
	}
	heap0[p]=a;
	heap1[p]=id;
}
inline void insert(double a,int id)
{
	heap0[++sz]=a;
	heap1[sz]=id;
	pushup(sz);
}
inline void Pop()
{
	heap0[1]=heap0[sz];
	heap1[1]=heap1[sz--];
	pushdown(1);
}
inline void Prime()
{
	int i;
	d[1]=0;
	insert(0.0,1);
	while(sz){
		int u=heap1[1];
		double w=heap0[1];
		Pop();
		if(vis[u]) continue;
		vis[u]=1;
		ans+=w;
		num++;
		if(num==n) break;
		f(i,1,n){
			if(vis[i]||i==u) continue;
			double tmp=cal(i,u);
			if(d[i]>tmp){
				d[i]=tmp;
				insert(d[i],i);
			}
		}
	}
}
int main()
{
	ios::sync_with_stdio(false);
	memset(d,127,sizeof(d));
	int i,j;
	cin>>n;
	f(i,1,n){
		cin>>a[i].x>>a[i].y;
	}
	Prime();
	cout<<fixed<<setprecision(2)<<ans<<endl;
	return 0;
}