HDU-1875 最小生成树 PRIM

最新推荐文章于 2019-04-08 18:27:44 发布

原创最新推荐文章于 2019-04-08 18:27:44 发布 · 475 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#dst #ini #graph #build

本文介绍了一个基于Prim算法实现的最小生成树问题解决方案。通过构建一个包含多个顶点的图，并利用距离矩阵来确定最短路径，实现了计算连接所有顶点的最小总权重。文章详细展示了如何使用C语言进行编程实现。

简单小题，没什么好说的。

/*
 * hdu-1875
 * mike-w
 * 2012-4-19
 */
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>
#define MAX_SIZE 128
#define INF (1e20)
#ifndef true
#define true (1)
#endif
#define EPS (1e-5)

double lake[MAX_SIZE][2];
double f[MAX_SIZE][MAX_SIZE];
double dst[MAX_SIZE];
int tag[MAX_SIZE];
int T,N;

int build_graph(void)
{
	int i,j;
	double dx,dy,dd;
	for(i=0;i<N;i++)
		for(j=i;j<N;j++)
		{
			dx=lake[i][0]-lake[j][0];
			dy=lake[i][1]-lake[j][1];
			dd=sqrt(dx*dx+dy*dy);
			if(dd>=10.0 && dd<=1000.0)
				f[i][j]=f[j][i]=dd*100.0;
			else
				f[i][j]=f[j][i]=INF;
		}
	return 0;
}

double prim(void)
{
	int i,x,fail;
	double cost;
	for(i=0;i<N;i++)
		tag[i]=0,dst[i]=INF;
	dst[0]=0;
	fail=0;
	cost=0.0;
	while(true)
	{
		/* extract mininal */
		x=-1;
		for(i=0;i<N&&tag[i];i++)
			;
		for(x=i++;i<N;i++)
			if(!tag[i] && dst[i]<dst[x])
				x=i;
		if(x>=N || fabs(dst[x]-INF)<EPS)
			break;
		/* update */
		tag[x]=1;
		cost+=dst[x];
		for(i=0;i<N;i++)
			if(!tag[i] && f[x][i]<INF && f[x][i]<dst[i])
				dst[i]=f[x][i];
	}
	for(i=0;i<N;i++)
		if(!tag[i])
			fail=1,i=N;
	return fail?-1.0:cost;
}

int main(void)
{
#ifndef ONLINE_JUDGE
	freopen("in","r",stdin);
#endif
	int i;
	double cost;
	scanf("%d",&T);
	while(T-->0)
	{
		scanf("%d",&N);
		for(i=0;i<N;i++)
			scanf("%lf%lf",lake[i],lake[i]+1);
		build_graph();
		cost=prim();
		if(cost<0)
			puts("oh!");
		else
			printf("%.1lf\n",cost);
	}
	return 0;
}