POJ 1679 The Unique MST

判断最小生成树是否唯一，有两种方法，见http://www.byvoid.com/blog/2-sp-mst/

我用的是删边的办法（Prim)

 

做完这道题发现有一个细节没搞清楚，就是now 只是用来更新松弛dis值（节点距离生成树的距离），当前加入生成树的边不一定有一个顶点为now

 

代码：

#include<iostream>using namespace std;const int MAX=105;const int inf=1<<30;int g[MAX][MAX],dis[MAX],slack[MAX];int n,m,e;struct node{int u,v;}mst[MAX];int prim(int flag){int i,j,min_weight,min_node,now,total=0;for(i=1;i<=n;i++){dis[i]=inf;}now=1;for(i=1;i<n;i++){min_weight=inf;dis[now]=-1;for(j=1;j<=n;j++){if(dis[j]>=0&&j!=now){if(dis[j]>g[now][j]){dis[j]=g[now][j];slack[j]=now;}if(dis[j]<min_weight){min_node=j;min_weight=dis[j];}}}if(min_weight==inf)//删边后MST不存在 return -1;if(flag){mst[i].u=min_node;mst[i].v=slack[min_node];//这里不能为now，因为当前min_node //不一定由当前now松弛 }total+=min_weight;now=min_node;}return total;}int secondMST(){int i,j,a,b,ans=inf;for(i=1;i<n;i++){///cout<<i<<endl;a=mst[i].u;b=mst[i].v;j=g[a][b];g[a][b]=g[b][a]=inf;ans=min(ans,prim(0));if(ans==-1)break;g[a][b]=g[b][a]=j;}return ans;}int main(){int i,j,a,b,c,T;scanf("%d",&T);while(T--){scanf("%d%d",&n,&m);for(i=1;i<=n;i++){for(j=1;j<=n;j++){if(i!=j)g[i][j]=inf;}}while(m--){scanf("%d%d%d",&a,&b,&c);g[a][b]=g[b][a]=c;}i=prim(1);//cout<<i<<endl;j=secondMST();//cout<<j<<endl;if(i==j)puts("Not Unique!");elseprintf("%d/n",i);}return 0;}