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; }