首页 > 代码库 > hdu1875浅谈prim算法的朴素实现

hdu1875浅谈prim算法的朴素实现

阅读原题

题目大意

给你几个(<=100)小岛的坐标,然后你把所有的岛都修上桥连接起来,求最小花费,还有个附加的限制:只有岛之间的距离大于等于10,或小于等于1000时才能修桥。

大概是因为十米以内不用建桥,千米以上无法建桥。哈哈,说着玩的。

很明显这是一道MST(最小生成树)的题目,貌似也有人用并查集AC过。

最小生成树算法


概述

最小生成树的常用算法有两个kruskal和prim算法。两者都是不停地执行归并操作,然而一言以蔽之,两者的不同之处在于:kruskal----归并边;prim----归并点

关于kruskal的题目原先AC过几道:

  • hdu1301
  • hdu1879
  • hdu3371

prim算法

先贴一个维基百科的链接“prim算法”。下面我用离散数学来描述一下。

设有图G=(V,E),所有的结点集合为V,另有一空集合U。基本思路是:

  1. 先随意确定一个起点。
  2. 设此点为v,加入集合U中。
  3. 观察与U中点相连的边,找到最短边。
  4. 把与最短边另一端的点加入到集合U中。
  5. 继续执行步骤3.知道V-U=0.即所有点都加入U中。

本题AC代码

朴素实现

这是最一般的实现方案。近乎模板的实现代码,在任一本数据结构书中可能都有提到。

void prim()
{
	double sum=0,lowcast[105]={0};
	int count=1;
	for(int i=0;i<c;i++)
	{
		lowcast[i] = d[0][i];
	}
	for(int i=0;i<c-1;i++)
	{
		double min = INF;//INF是自定义常量,表无穷大
		int k = 105;
		for(int j=0;j<c;j++)
			if(min>lowcast[j]&&lowcast[j])
			{
				min = lowcast[j];
				k = j;
			}
			if(k!=105)
			{
				sum+=lowcast[k];
				lowcast[k]=0;
				count++;
			}
			for(int j=0;j<c;j++)
				if(d[j][k]<lowcast[j])
				{
					lowcast[j] = d[j][k];
				}
	}
	if(count==c)
		printf("%.1lf\n",sum*100);
	else
		printf("oh!\n");
}

解释

二维数组d,保存着所有岛两两的距离。首先我们选择的起点是结点0.

for(int i=0;i<c;i++)
{
	lowcast[i] = d[0][i];
}
lowcast数组保存的是集合U中的点到V-U中每个结点的最近距离。一开始的时候,集合U中只有结点0,所以lowcast数组保存的都是结点0到每个结点的距离。

for(int i=0;i<c-1;i++)
{
	double min = INF;//INF是自定义常量,表无穷大,实际不可能为无穷大
	int k = 105;//初始化为105,因为实际不可能为105
这是prim算法的主要部分开始了,初始化min用于保存到集合U中的点最小距离的边的长度。k就是用于保存最短边的另一端的结点名称。

	for(int j=0;j<c;j++)
		if(min>lowcast[j]&&lowcast[j])
		{
			min = lowcast[j];
			k = j;
		}
这里也比较好理解,就是在不停的更新最短边的长度及最短边的另一端结点名。注意lowcast[j]==0时,表示结点 j 已经加入集合U了。

	if(k!=105)
	{
		sum+=lowcast[k];
		lowcast[k]=0;
		count++;
	}
如果k!=105,表示确实找到了最短边。所以把这条边的长度lowcast[k]加入到总和sum中,同时标记该点k已经加入集合U,即lowcast[k]=0。count用于记录已加入集合U中的点的个数。

	for(int j=0;j<c;j++)
		if(d[j][k]<lowcast[j])
		{
			lowcast[j] = d[j][k];
		}
因为集合U中的点可能已经增加了,所以集合U中的点到V-U的点的最近距离也要再次更新。这个操作被称为 “松弛操作”。在图论问题中有较多应用。

完整AC代码

#include <iostream>
using namespace std;
#include <cmath>
const double INF=0x3f3f3f3f*1.0;
struct node
{
	int x;
	int y;
};

double d[105][105];
int c;//岛屿个数
void prim()
{
	double sum=0,lowcast[105]={0};
	int count=1;
	for(int i=0;i<c;i++)
	{
		lowcast[i] = d[0][i];
	}
	for(int i=0;i<c-1;i++)
	{
		double min = INF;//INF是自定义常量,表无穷大
		int k = 105;
		for(int j=0;j<c;j++)
			if(min>lowcast[j]&&lowcast[j])
			{
				min = lowcast[j];
				k = j;
			}
		if(k!=105)
		{
			sum+=lowcast[k];
			lowcast[k]=0;
			count++;
		}
		for(int j=0;j<c;j++)
			if(d[j][k]<lowcast[j])
			{
				lowcast[j] = d[j][k];
			}
	}
	if(count==c)
		printf("%.1lf\n",sum*100);
	else
		printf("oh!\n");
}
int main()
{

	int t;
	cin>>t;
	while(t--)
	{
		node n[105];
		cin>>c;
		for(int i=0;i<c;i++)
			cin>>n[i].x>>n[i].y;
		for(int i=0;i<c;i++)
			for(int j=0;j<c;j++)
			{
				if(i==j)
				{		
					d[i][j]=0;
					continue;
				}

				int x1=n[i].x, x2=n[j].x;
				int y1=n[i].y, y2=n[j].y;
				double dist=sqrt(1.0*(x1-x2)*(x1-x2)+(y1-y2)*(y1-y2));
				if(dist<10.0||dist>1000.0)
					d[j][i]=d[i][j]=INF;
				else
					d[j][i]=d[i][j]=dist;
			}
			prim();
	}
	return 0;
}

其他注释

const double INF=0x3f3f3f3f*1.0;
自定义表示无穷大的double常量。关于这个值的选取可以参考这篇文章《编程中无穷大常量的设定技巧》。

在计算岛与岛之间距离的时候。

    if(i==j)
    {        
        d[i][j]=0;
        continue;
    }
表示如果是i==j,就直接设为0,在prim算法里对应lowcast也会是0,会被忽略掉。我这里不能设成INF,我的实现方案中INF会导致AC不了的。大家具体方案具体分析啦。