【SSL】1618剑鱼行动

【SSL】1618剑鱼行动

Time Limit:1000MS
Memory Limit:65536K

Description

给出N个点的坐标，对它们建立一个最小生成树，代价就是连接它们的路径的长度，现要求总长度最小。N的值在100以内，坐标值在[-10000,10000].结果保留二位小数

Input

5 //---------------5个点
0 0 //---------------5个点点的坐标
0 1
1 1
1 0
0.5 0.5

Output

2.83

Sample Input

5 
0 0 
0 1
1 1
1 0
0.5 0.5

Sample Output

2.83

思路

用最小生成树，先算出点两两间的距离。再用prim或kruskal。

代码

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
int n,tot=0;
double a[110][110],mn[110],x[110],y[110],ans=0;
bool d[110];
void input()
{
	int i,j;
	scanf("%d",&n);
	mn[0]=100000000;
	for(i=1;i<=n;mn[i]=100000000,i++)
		scanf("%lf%lf",&x[i],&y[i]);
	for(i=1;i<n;i++)
		for(j=i+1;j<=n;j++)
			a[i][j]=a[j][i]=sqrt(abs(x[i]-x[j])*abs(x[i]-x[j])+abs(y[i]-y[j])*abs(y[i]-y[j]));//计算距离
	return;
}
void prim()
{
	int i,j,w;
	memset(d,true,sizeof(d));
	mn[1]=0;
	for(i=1;i<n;i++)
	{
		for(w=0,j=1;j<=n;j++)//找最近蓝点
			if(d[j]&&mn[j]<mn[w])
				w=j;
		d[w]=0;
		ans+=mn[w];
		for(j=1;j<=n;j++)//修改最短距离
			if(d[j]&&a[w][j]<mn[j])
				mn[j]=a[w][j];
	}
	return;
}
int main()
{
	input();
	prim();
	printf("%.2lf",ans);
	return 0;
}