2019 GDUT 新生专题Ⅲ I题

最新推荐文章于 2020-12-16 16:51:43 发布

Jhaogee

最新推荐文章于 2020-12-16 16:51:43 发布

阅读量187

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/weixin_45794203/article/details/104010202

本文介绍了一种解决连通所有节点最短路径问题的方法，使用Prim算法构建最小生成树。通过具体代码示例，详细解释了Prim算法的实现过程，适用于解决网络设计、电路布局等问题。

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

I - 还是畅通工程

题目链接
题目大意：计算能连通所有村庄的最短路线。
思路：直接套最小生成树模板prim或者Kruskal。下面代码用了prim。
代码

#include <cstdio>
#include <algorithm>
using namespace std;
const int MAX=105;
const int inf=999999999;
int mincost[MAX],cost[MAX][MAX],n;
bool used[MAX];
int prim(){
	mincost[1]=0;
	int res=0;
	while(true){
		int v=-1;
		for(int i=1;i<=n;i++){
			if(!used[i]&&(v==-1||mincost[i]<mincost[v])) v=i;
		}
		if(v==-1) break;
		used[v]=true;
		res+=mincost[v];
		for(int i=1;i<=n;i++)
		mincost[i]=min(mincost[i],cost[v][i]);
	}
	return res;
}
int main(){
	int u,v,d;
	while(scanf("%d",&n)==1&&n){
		int m=(n-1)*n/2;
		fill(cost[0],cost[0]+MAX*MAX,inf);
		fill(mincost,mincost+MAX,inf);
		fill(used,used+MAX,false);
		for(int i=1;i<=n;i++) cost[i][i]=0;
		for(int i=0;i<m;i++){
			scanf("%d%d%d",&u,&v,&d);
			cost[u][v]=cost[v][u]=d;
		}
		int res=prim();
		printf("%d\n",res);
	}
	return 0;
}