题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1863
源代码:
//HDOJ 1863 AC
#include <iostream>
using namespace std;
#define MAX 99999
#define LEN 101//要开到101以上 100不知道为什么TLE?
int dist[LEN];
int map[LEN][LEN];
bool isvisited[LEN];
int m;
void init(){
int i,j;
for(i=0;i<LEN;i++){
for(j=0;j<LEN;j++){
map[i][j]=MAX;
}
}
}
//prime算法
int prime(int n){
int i,j,min,pos,sum;
sum=0;
memset(isvisited,false,sizeof(isvisited));
//初始化
for(i=1;i<=n;i++){
dist[i]=map[1][i];
}
//从1开始
isvisited[1]=true;
dist[1]=MAX;
//找到权值最小点并记录下位置
for(i=1;i<n;i++){
min=MAX;
//pos=-1;
for(j=1;j<=n;j++){
if(!isvisited[j] && dist[j]<min){
min=dist[j];
pos=j;
}
}
// if(pos==-1){ //两种判断方式都会AC
// return -1;
// }
if(min==MAX){
return -1;
}
sum+=min;
isvisited[pos]=true;
//更新权值
for(j=1;j<=n;j++){
if(!isvisited[j] && dist[j]>map[pos][j]){
dist[j]=map[pos][j];
}
}
}
return sum;
}
int main(){
int n,m;
while(scanf("%d%d",&n,&m),n){
init();
int i,a,b,w,ans;
if(n==0) break;
for(i=0;i<n;i++){
cin>>a>>b>>w;
if(map[a][b]>w){
map[a][b]=map[b][a]=w;
}
}
ans=prime(m);
if(ans==-1){
puts("?");
}
else {
cout<<ans<<endl;
}
}
return 0;
}