求次小的生成树即求第二小的生成树,次小生成树可由最小生成树换一条边得到,一般采用的是求出最小生成树后,依次删除最小生成树上的每一条边,然后生成n-1个最小生成树,记录下这个过程中的最小生成树的值,那么这个就是第二小生成树了,用kruskal这种算法的复杂度为O(n*elog2e),当图比较稠密时,复杂度接近O(n^3)
次小生成树用kruskal的第一种方法
#include<bits/stdc++.h>
using namespace std;
int n,m;
struct data
{
int a,b,w;
bool vis;
} p[20010];
vector<int>g[110];
int father[110],len[110][110];
const int oo=1e9;
bool cmp(data a,data b)
{
if(a.w!=b.w)
return a.w<b.w;
if(a.a!=b.a)
return a.a<b.a;
return a.b<b.b;
}
int Union_Find(int x)//并查集
{
if(x!=father[x])
return father[x]=Union_Find(father[x]);
return father[x];
}
void kruskal()
{
sort(p,p+m,cmp);
for(int i=0; i<=n; i++)//初始化
{
g[i].clear();
g[i].push_back(i);
father[i]=i;
}
int sum=0,k=0;//sum是最小生成树的值
for(int i=0; i<m; i++)
{
if(k==n-1) break;
int x1=Union_Find(p[i].a);
int x2=Union_Find(p[i].b);
if(x1!=x2)
{
k++;
p[i].vis=1;
sum+=p[i].w;
int l=g[x1].size();
int ll=g[x2].size();
for(int j=0; j<l; j++)
for(int k=0; k<ll; k++)
len[g[x1][j]][g[x2][k]]=len[g[x2][k]][g[x1][j]]=p[i].w;
father[x1]=x2;
int tem[110];
for(int j=0; j<ll; j++)
tem[j]=g[x2][j];
for(int j=0; j<l; j++)
g[x2].push_back(g[x1][j]);
for(int j=0; j<ll; j++)
g[x1].push_back(tem[j]);
}
}
int cisum=oo;//次小生成树的权值
for(int i=0; i<m; i++)
if(!p[i].vis)
cisum=min(cisum,sum+p[i].w-len[p[i].a][p[i].b]);
if(cisum>sum)
printf("cisum=%d\n",cisum);
}
int main()
{
while(~scanf("%d%d",&n,&m))
{
for(int i=0; i<m; i++)
{
scanf("%d%d%d",&p[i].a,&p[i].b,&p[i].w);
}
kruskal();
}
return 0;
}
kruskai
#include<stdio.h>
#include<bits/stdc++.h>
using namespace std;
const int maxn=10000+10;
struct data
{
int x,y,w;
} sides[maxn];
bool flag[maxn];
int sett[maxn];
int cmp(data a,data b)
{
return a.w<b.w;
}
int Find(int x)
{
if(x==sett[x])
return x;
sett[x]=Find(sett[x]);
return sett[x];
}
int main()
{
int n,m,ans,ans1,rx,ry,i,j,before,t,k;
bool ok;
while(~scanf("%d %d",&n,&m))
{
ok=0;
for(i=0; i<m; i++)
scanf("%d%d%d",&sides[i].x,&sides[i].y,&sides[i].w);
sort(sides,sides+m,cmp);
for(i=1; i<=n; i++)sett[i]=i;
memset(flag,0,sizeof(flag));
ans=0;
int cnt=1;
for(i=0; i<m; i++)
{
rx=Find(sides[i].x);
ry=Find(sides[i].y);
if(rx!=ry)
{
cnt++;
ans+=sides[i].w;
sett[rx]=ry;
flag[i]=1;
if(cnt==n)
break;
}
}
before=1;
flag[before]=0;
for(i=1; i<=n-1; i++)
{
for(k=1; k<=n; k++)
sett[k]=k;
ans1=0;
cnt=1;
for(j=0; j<m; j++)
{
if(j!=before)
{
rx=Find(sides[j].x);
ry=Find(sides[j].y);
if(rx!=ry)
{
cnt++;
ans1+=sides[j].w;
sett[rx]=ry;
if(cnt==n)
break;
}
}
}
if(ans1==ans)
{
ok=1;
break;
}
for(t=before+1; t<m; t++)
{
if(flag[t])
{
before=t,flag[before]=0;
break;
}
}
}
if(!ok)
printf("ans1=%d\n",ans1);
}
return 0;
}
有一种更为简单的方法:先求最小生成树T,枚举添加不在T中的边,则添加后一定会形成环,找到环上边值第二大的边,把它删掉,计算当前生成树的权值,取所有枚举修改的生成树的最小值,即为次小生成树。这种方法的实现更为简单,首先求最小生成树T,然后从每个结点u,遍历最小生成树T,用一个二维的数组max[u][v]记录结点u到结点v的路径上边的最大值,然后枚举不在T中的边(u,v),计算T-max[u][v]+w(u,v)的最小值,即为次小生成树的权值 ,这种方法的时间复杂度为O(n^2+e)
#include<bits/stdc++.h>
using namespace std;
const int inf=0x3f3f3f3f;
const int maxn=100+10;
bool link[maxn][maxn],vis[maxn];
int w[maxn][maxn],lowc[maxn],pre[maxn],Max[maxn][maxn];
int n,m;
int prim()
{
int i,j,p,k;
int minc,res=0;
memset(vis,false,sizeof(vis));
memset(pre,0,sizeof(pre));
memset(Max,0,sizeof(Max));
vis[1]=true,pre[1]=1;
for(i=2; i<=n; i++) //初始化
{
lowc[i]=w[1][i];
pre[i]=1;
}
for(i=2; i<=n; i++) //prim
{
minc=inf,p=-1;
for(j=1; j<=n; j++)
{
if(!vis[j]&&lowc[j]<minc)
{
minc=lowc[j];
p=j;
}
}
vis[p]=true;
res+=minc;//最小生成树加权值
Max[pre[p]][p]=minc;
link[pre[p]][p]=true;//将这两条边标记为最小生成树的边
link[p][pre[p]]=true;
for(k=1; k<=n; k++)
Max[k][p]=max(Max[pre[p]][p],Max[k][p]);
for(j=1; j<=n; j++)
if(!vis[j]&&lowc[j]>w[p][j])
{
lowc[j]=w[p][j];
pre[j]=p;
}
}
return res;
}
int main()
{
// freopen("in.txt","r",stdin);
// freopen("out.txt","w",stdout);
int s,e,t,ans,ans1;
while(~scanf("%d%d",&n,&m))
{
int i,j;
bool ok=true;//是否唯一最小生成树的标志
for(i=1; i<=n; i++)
for(j=1; j<=n; j++)
w[i][j]=inf;
memset(link,false,sizeof(link));
for(i=1; i<=m; i++)
{
scanf("%d%d%d",&s,&e,&t);
w[s][e]=t;
w[e][s]=t;
}
ans=prim();//最小生成树的权值
for(i=1; i<=n; i++)
{
for(j=i+1; j<=n; j++)
{
if(w[i][j]!=inf&&!link[i][j])
{
ans1=ans+w[i][j]-Max[i][j];//ans1次小生成树的权值
}
if(ans1==ans)
{
ok=0;
break;
}
}
if(!ok)
break;
}
printf("ans=%d ans1=%d\n",ans,ans1);
}
return 0;
}