HDU--1683--prim算法

省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可）。经过调查评估，得到的统计表中列出了有可能建设公路的若干条道路的成本。现请你编写程序，计算出全省畅通需要的最低成本。

Input

测试输入包含若干测试用例。每个测试用例的第1行给出评估的道路条数 N、村庄数目M ( < 100 )；随后的 N
行对应村庄间道路的成本，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间道路的成本（也是正整数）。为简单起见，村庄从1到M编号。当N为0时，全部输入结束，相应的结果不要输出。

Output

对每个测试用例，在1行里输出全省畅通需要的最低成本。若统计数据不足以保证畅通，则输出“?”。

Sample Input
3 3
1 2 1
1 3 2
2 3 4
1 3
2 3 2
0 100
Sample Output
3
?

思路：基础图论，最小生成树；prim()算法即可解决问题；

#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int MAX=10100; 
const int INF=0x3f3f3f3f;
int cost[MAX][MAX];
int mincost[MAX];
bool used[MAX]; 
int V;
int prim(){
	memset(mincost,0,sizeof(mincost));
	for(int i=1;i<=V;i++){
		mincost[i]=INF;
		used[i]=false;
	}
	mincost[1]=0;
	int res=0;
	while(true){ 
		int v=-1;
		for(int u=1;u<=V;u++){
			if(!used[u]&&(v==-1||mincost[u]<mincost[v]))
				v=u;
		}
		if(v==-1)
			break;
		used[v]=true;
		res+=mincost[v];
		for(int u=1;u<=V;u++){
			mincost[u]=min(mincost[u],cost[v][u]);
		}
	}
	return res;
}
void init(){
	for(int i=1;i<=V;i++)
		for(int j=1;j<=V;j++)
			if(i==j)	cost[i][j]=0;
			else 	cost[i][j]=INF;
}
int main(){
	int n,m;
	while(1){
		scanf("%d%d",&n,&m);
		if(n==0)	break;
		V=m;
		init();
		int e1,e2,costa;
		for(int i=1;i<=n;i++){ 
			scanf("%d%d%d",&e1,&e2,&costa);
			cost[e1][e2]=costa;
		} 
		int ans=prim();
		if(ans>=INF)	printf("?\n");
		else	printf("%d\n",ans);
	}
}