poj 1679 The Unique MST

求次小生成树，下面的两篇文章写得很好

http://www.docin.com/p-90421675.html

http://blog.csdn.net/sunbaigui/archive/2009/10/11/4653271.aspx

核心思想：先用prim求出最小生成树，在求最小生成树的过程中，求出连通任意两个节点u，i的路径上的最长边，记录在len[u][i]中，

len[u][pi[u]]=len[pi[u]][u]=cost[pi[u]][u];

for(i=1;i<=n;i++)                        
            if(s[i])
                len[i][u]=len[u][i]=mymax(len[i][pi[u]],len[pi[u]][u]);

#include<iostream> using namespace std; #define N 105 #define max_int 99999999 int cost[N][N],dist[N],s[N],len[N][N],pi[N]; int choose(int n) { int pos=-1,m=max_int; for(int i=1;i<=n;i++) if(!s[i]&&dist[i]<m) { m=dist[i]; pos=i; } return pos; } int mymax(int a,int b) { if(a>b) return a; else return b; } int next_prim(int n) { int u,i,weight=0; for(i=1;i<=n;i++)//求最小生成树时的预处理 { dist[i]=max_int; s[i]=0; len[i][i]=0; } dist[1]=0; s[1]=1; pi[1]=1; for(i=2;i<=n;i++) if(dist[i]>cost[1][i]) { dist[i]=cost[1][i]; pi[i]=1; } while(1) { u=choose(n); if(u==-1) break; len[u][pi[u]]=len[pi[u]][u]=cost[pi[u]][u];//在求最小生成树的过程中，求出任意两个节点的路径上的最大边 for(i=1;i<=n;i++) //动态规划 if(s[i]) len[i][u]=len[u][i]=mymax(len[i][pi[u]],len[pi[u]][u]); s[u]=1; weight+=dist[u]; for(i=1;i<=n;i++) if(!s[i]&&dist[i]>cost[u][i]) { pi[i]=u; dist[i]=cost[u][i]; } } return weight; } int main() { int ca,n,m,i,j,xi,yi,di,mst,unique; scanf("%d",&ca); while(ca--) { scanf("%d%d",&n,&m); for(i=1;i<=n;i++) for(j=i;j<=n;j++) cost[i][j]=cost[j][i]=max_int; for(i=1;i<=m;i++) { scanf("%d%d%d",ξ,&yi,&di); cost[xi][yi]=cost[yi][xi]=di; } mst=next_prim(n); unique=max_int; for(i=1;i<=n;i++)//枚举不在树中的边，求次小生成树 for(j=i+1;j<=n;j++) if(pi[j]!=i&π[i]!=j&&cost[i][j]<max_int) if(unique>mst-len[i][j]+cost[i][j]) unique=mst-len[i][j]+cost[i][j]; if(unique==mst) printf("Not Unique!/n"); else printf("%d/n",mst); } return 0; }