hdu 1863 畅通工程

#include<stdio.h>
#include<string.h>
#define inf 100000
int map[110][110],dis[110];
int n,vis[110];
int Prim(){
	int i,j,k,tmp,ans;
	for(i=1;i<=n;i++)
	dis[i]=inf;//初始化 
	dis[1]=0;
	for(i=1;i<=n;i++){
		tmp=inf;
		for(j=1;j<=n;j++){
			if(!vis[j]&&tmp>dis[j]){
				tmp=dis[j];
				k=j;
			}//找出最小的顶点 
		}
		vis[k]=1;
		for(j=1;j<=n;j++){
			if(!vis[j]&&dis[j]>map[k][j])
			dis[j]=map[k][j];//更新距离 
		}
	}
	for(ans=0,i=1;i<=n;i++){
		if(dis[i]==inf) break;
		ans+=dis[i];
	}
	if(i<=n) return -1;
	else return ans;
		
}//Prim算法模板
int main(){
	int m,i,j,a,b,val,s;
	while(scanf("%d%d",&m,&n),m){
		memset(vis,0,sizeof(vis));
		for(i=1;i<=n;i++)
		for(j=1;j<=n;j++){
			if(i==j) map[i][j]=0;
			else map[i][j]=inf;
			}
		for(i=0;i<m;i++){
			scanf("%d%d%d",&a,&b,&val);
			if(map[a][b]>val)
			map[a][b]=map[b][a]=val;
			}
			s=Prim();
		if(s==-1) printf("?\n");
		else printf("%d\n",s);	
		}
	return 0;
}

你可能感兴趣的:(Prim)