洛谷 [P1265] 公路修建

本题的描述：城市联盟，最短距离。。

使人想到了prim求MST，再一看数据范围：完全图！，那么一定得用prim，因为只有5000个点，所以不加优化的prim就能过。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <cmath>
using namespace std;
const int MAXN=5005;
int init(){
	int rv=0,fh=1;
	char c=getchar();
	while(c<'0'||c>'9'){
		if(c=='-') fh=-1;
		c=getchar();
	}
	while(c>='0'&&c<='9'){
		rv=(rv<<1)+(rv<<3)+c-'0';
		c=getchar();
	}
	return rv*fh;
}
int n,x[MAXN],y[MAXN];
double dis[MAXN],tot;
bool f[MAXN];
double cal(int a,int b){
	double r1=x[a]-x[b],r2=y[a]-y[b];
	return sqrt(r1*r1+r2*r2);
}
int main(){
	freopen("in.txt","r",stdin);
	n=init();
	for(int i=1;i<=n;i++){
		x[i]=init();y[i]=init();
	}
	memset(dis,0x7f,sizeof(dis));
	dis[1]=0.0;
	f[1]=1;
	for(int i=2;i<=n;i++){
		dis[i]=cal(1,i);
	}
	for(int k=2;k<=n;k++){
		double mi=1e20;
		int t=0;
		for(int i=1;i<=n;i++){
			if(!f[i]){
				if(dis[i]<mi) {mi=dis[i];t=i;}
			}
		}
		if(t==0) break;
		tot+=mi;
		f[t]=1;
		for(int i=1;i<=n;i++){
			if(!f[i]){
				dis[i]=min(dis[i],cal(t,i));
			}//在这里写的时候有一种dijkstra的感觉，但要注意这里的dis[]表示的是到已生成的MST的距离，
                         不是到某一点的距离。
		}
	}
	printf("%.2lf",tot);
	fclose(stdin);
	return 0;
}

noip完了之后要学堆优化prim。。。。