hdoj 1863 畅通工程 最小生成树---prime算法

题目： http://acm.hdu.edu.cn/showproblem.php?pid=1863

注意有可能出现无法生成树的情况。

#include <iostream>#include <cstring>using namespace std;const int inf=0xffffff;int weight[105][105],n,m;int prime(){int sum=0;int pre[105];int dis[105];bool visit[105];memset(visit,0,sizeof(visit));for(int i=1;i<=m;i++){dis[i]=weight[1][i];pre[i]=1;}visit[1]=1;for(int i=1;i<m;i++){int minimum=inf;int index=-1;for(int j=1;j<=m;j++){if(!visit[j] &&dis[j]<minimum){minimum=dis[j];index=j;}}if(index==-1)return inf;visit[index]=1;sum+=weight[pre[index]][index];for(int j=1;j<=m;j++){if(!visit[j]&& dis[j]>weight[index][j]){dis[j]=weight[index][j];pre[j]=index;}}}return sum;} int main(){while(cin>>n>>m){if(n==0)break;for(int i=1;i<=m;i++)for(int j=1;j<=m;j++)weight[i][j]=inf;for(int i=0;i<n;i++){int x,y,z;cin>>x>>y>>z;weight[x][y]=z;weight[y][x]=z;}int ans=prime();if(ans==inf){cout<<"?"<<endl;continue;}cout<<ans<<endl;}}