HDU 1223 还是畅通工程(最小生成树prim模板)

一个很简单的prim模板,但虽然是模板,但也是最基础的,也要脱离模板熟练打出来

后期会更新kruskal写法

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn = 1000;
const int INF = 0x3f3f3f3f;//int型最大值 
int Map[maxn][maxn],vis[maxn];
int n,dis[maxn],ans;

void init(){
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			if(i==j)Map[i][j]=0;
			else Map[i][j]=INF;
		}
	}
	ans=0;
	memset(vis,0,sizeof(vis));
}

void input(){
	int u,v,w;
	for(int i=0;i<n*(n-1)/2;i++){
		scanf("%d%d%d",&u,&v,&w);
		if(Map[u][v]>w)
		Map[u][v]=Map[v][u]=w;
	}
}

int prim(){
	for(int i=1;i<=n;i++){
		dis[i]=Map[1][i];
	}
	vis[1]=1;
	for(int i=1;i<n;i++){
		int MIN=INF,x=-1;
		for(int j=1;j<=n;j++){
			if(!vis[j]&&dis[j]<MIN){//每一次都取该结点和其他相邻的结点的最小值 
				MIN=dis[j];
				x=j;
			}
		}
		if(x==-1)return -1;
		ans+=MIN;
		vis[x]=1;//标记已经过的节点 
		for(int j=1;j<=n;j++){
			if(!vis[j]&&dis[j]>Map[x][j]){//更新dis数组, 
				dis[j]=Map[x][j];
			}
		}
	}
}

int main(){
	while(scanf("%d",&n)&&n){
		init();//初始化 
		input();//输入 
		prim();//关键 
 		printf("%d\n",ans);
	}
	return 0;
} 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值