HDOJ 1863 畅通工程

题目链接：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;

}