题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1301
源代码:
//HDOJ1301
#include<iostream>
using namespace std;
#define MAX 99999
#define LEN 30
int dist[LEN];//某点的权值 起始点到目标点的权值
int map[LEN][LEN];//某点到某点两点之间的权值
bool isvisitd[LEN];//表示某点是否访问过
//初始化map数组 设置为无穷大
void init(int n){
for(int i = 0;i<=n;i++) {
for(int j = 0;j<=n;j++)
map[i][j] = MAX;
}
}
//prim最小生成树的算法
int prim(int n){
int i,j,min,pos,sum;
sum = 0; //最小生成树的权值
//初始化,表示没有一点走过
memset(isvisitd,false,sizeof(isvisitd));
//初始化给dist数组赋值
for(i = 1;i<=n;i++){
dist[i] = map[1][i];
}
isvisitd[1] = true;//标记1已被访问 从1开始
//找到权值最小点并记下位置
for(i = 1;i<n;i++){
min = MAX;
for(j = 1;j<=n;j++){
if(!isvisitd[j]&&dist[j]<min){
min = dist[j];
pos = j;//记录下该位置
}
}
sum+=min;
isvisitd[pos] = true;
//更新权值
for(j = 1;j<=n;j++) {
if(!isvisitd[j]&&dist[j]>map[pos][j]){
dist[j] = map[pos][j];
}
}
}
return sum;
}
int main(){
int i,j,n,m,len;
char start,end;
while(scanf("%d",&n)!=EOF){
if(n==0){
break;
}
init(n);//初始化
for(i=0;i<n-1;i++){
cin>>start>>m;
for( j = 0;j<m;j++){
cin>>end>>len;
map[i+1][end-'A'+1] = len;
map[end-'A'+1][i+1] = len;
}
}
cout<<prim(n)<<endl;
}
return 0;
}