次小生成树的求法:
1.Prime法
定义一个二维数组F[i][j]表示点i到点j在最小生成树中的路径上的最大权值。有个知识就是将一条不在最小生成树中的边Edge加入最小生成树时,树中要去掉的边就是Edge连接的两个端点i,j的F[i][j]。这样就能保存找到的生成树时次小生成树。
代码如下:
#include<iostream> #include<algorithm> #include<cstdio> #include<cstring> #include<cmath> #define inf 1<<30 #define Maxn 102 #define Maxm 10010 #define USE 2 #define EXIST 1 #define NOTEXIST 0 using namespace std; int map[Maxn][Maxn],dist[Maxn],vi[Maxn],f[Maxn][Maxn],use[Maxn][Maxn],pre[Maxn]; int n,m; int prime(int src) { int i,j,Min,index; int ans=0; memset(vi,0,sizeof(vi)); memset(pre,-1,sizeof(pre)); for(i=1;i<=n;i++) dist[i]=inf;//一定要初始化为inf,这样以第一个点开始,使与第一个相连的节点的前节点为第一个节点。 dist[1]=0;//以第一个节点开始 for(i=1;i<=n;i++) { Min=inf; for(j=1;j<=n;j++) { if(!vi[j]&&dist[j]<Min) { Min=dist[j]; index=j; } } if(pre[index]!=-1)//如果存在前节点 { use[index][pre[index]]=use[pre[index]][index]=USE;//标记为使用过 for(j=1;j<=n;j++) if(vi[j])//对树种已存在的点进行更新 f[j][index]=max(f[j][pre[index]],map[index][pre[index]]); } ans+=Min; vi[index]=1; //cout<<Min<<"*"<<endl; for(j=1;j<=n;j++) { if(!vi[j]&&dist[j]>map[index][j]) { dist[j]=map[index][j]; pre[j]=index; } } } //cout<<ans<<"*"<<endl; return ans; } int secondmst(int mst) { int i,j,ans; ans=inf; for(i=1;i<=n;i++) for(j=1;j<=n;j++) if(use[i][j]==EXIST) { if(mst+map[i][j]-f[i][j]<ans)//求次小生成树 ans=mst+map[i][j]-f[i][j]; } //cout<<ans<<"*"<<endl; return ans; } void init()//初始化 { int i,j; memset(f,0,sizeof(f)); for(i=1;i<=Maxn-1;i++) for(j=1;j<=Maxn-1;j++) map[i][j]=map[j][i]=inf; memset(use,0,sizeof(use)); } int main() { int i,j,a,b,c,t; scanf("%d",&t); while(t--) { scanf("%d%d",&n,&m); init(); for(i=1;i<=m;i++) { scanf("%d%d%d",&a,&b,&c); map[a][b]=map[b][a]=c; use[a][b]=use[b][a]=1; } int ans1=prime(1); int ans2=secondmst(ans1); if(ans1==ans2) printf("Not Unique!\n"); else printf("%d\n",ans1); } return 0; }