HDU1102,Constructing Roads(Prim算法)

Prim算法适用于稠密图,关于其详解,可参考博客:
https://blog.csdn.net/yeruby/article/details/38615045
代码如下:

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
int map[105][105],dis[105];	//dis[]记录当前连通块与其它未连接的节点的最小距离
bool vis[105];	//vis表示哪个节点已经加入连通块了,相当于用数组来表示一个集合

int Prim(int n)
{
	int ans=0;
	memset(vis,0,sizeof(vis));
	for(int i=1;i<=n;i++)	
		dis[i]=map[1][i];	//任选某一节点作为起点,在这里选1
	vis[1]=1;	//1加入集合
	for(int i=1;i<=n;i++)	//注意,这里的i并不是表示节点编号,只是一个循环的标记,没有什么太多的意义,可以不理它
	{
		int temp=1005,k;	//temp用来探测dis[]中最小的距离,而k记录最小距离对应的节点编号
		for(int j=1;j<=n;j++)	//遍历n个节点
			if(!vis[j]&&temp>dis[j])	//前提:vis[j]=0,即这个节点还未被加入连通块
			{
				temp=dis[j];
				k=j;
			}
		if(temp==1005)	//如果temp不变,说明所以节点都连通了,就可以结束了
			break;
		ans+=dis[k];
		vis[k]=1;	//将k加入集合
		for(int j=1;j<=n;j++)	//要更新dis[]
			if(!vis[j]&&dis[j]>map[k][j])
				dis[j]=map[k][j];
	}
	return ans;
}

int main()
{
	int n,q;
	while(cin>>n)
	{
		for(int i=1;i<=n;i++)
			for(int j=1;j<=n;j++)
				cin>>map[i][j];
		cin>>q;
		while(q--)
		{
			int a,b;
			cin>>a>>b;
			map[a][b]=map[b][a]=0;	//一开始,我还在后面添加vis[a]=vis[b]=1,因为当时认为既然他们已经连通了,就应该把a,b加入集合
			//这么理解好像没问题,但是因为算法编程实现的原因,这样处理最终会出错,此时我们应将map[a][b]=map[b][a]=0理解为,a与b的距离为0,或连通代价为0
		}
		cout<<Prim(n)<<endl;
	}
	return 0;
 } 
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值