poj2421 Constructing Roads

最新推荐文章于 2020-02-07 11:05:28 发布

turbo_7

最新推荐文章于 2020-02-07 11:05:28 发布

阅读量588

点赞数

CC 4.0 BY-SA版权

分类专栏：最小生成树 poj 文章标签：最小生成树 POJ prim

本文链接：https://blog.youkuaiyun.com/turbo_7/article/details/8963106

poj 同时被 2 个专栏收录

39 篇文章

订阅专栏

最小生成树

4 篇文章

订阅专栏

本文介绍了一种改进的Prim算法实现方式，通过初始化已建成路径的边权为0来优化最小生成树的计算过程。文章详细展示了算法的具体实现代码，并讨论了如何处理自环的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

可以将已经建好的路初始化为路程为0（这样在进行prim算法求最小生成树时已经建好的路会被首先加入进去，并且不会对我们所要求的还需建设的路径长度有贡献），那么为了避免混淆，到达自身的路程就不能再做初始化为0处理了，这里全都初始化为无穷大，剩下的就是朴素的prim算法了。

#include<iostream>
#include<cstring>
#include<cstdio>
#define INF 0x3f3f3f3f
int ma[105][105];
int cost[105];
int n,m;
int prim(int s){
	int i,j;
	for(i=0;i<n;i++)
		cost[i]=ma[s][i];
	cost[s]=-1;
	int sum=0;
	int mini;
	for(i=0;i<n-1;i++){
		mini=INF;
		for(j=0;j<n;j++)
			if(cost[j]<mini&&cost[j]!=-1){
				s=j;
				mini=cost[j];
			}
		sum+=mini;
		cost[s]=-1;
		for(j=0;j<n;j++)
			if(cost[j]>ma[s][j])
				cost[j]=ma[s][j];
	}
	return sum;
}
int main(){
	int i,j;
	int a,b;
	while(scanf("%d",&n)!=EOF){
		for(i=0;i<n;i++)
			for(j=0;j<n;j++){
				scanf("%d",&ma[i][j]);
				if(i==j)
					ma[i][i]=INF;
			}
			scanf("%d",&m);
			for(i=0;i<m;i++){
				scanf("%d%d",&a,&b);
				--a;
				--b;
				ma[a][b]=ma[b][a]=0;
			}
			printf("%d\n",prim(0));
	}
	return 0;
}