定义
对于一个图中,第一个比最小生成树边权和小的生成树叫次小生成树。
思路
先跑最小生成树(Kruskal),用了的 v i s i ← 1 vis_i \gets 1 visi←1,然后遍历到 m m m,如果 v i s i = 1 vis_i = 1 visi=1,设为 2 2 2,搞一个新的 Kruskal, v i s i = 2 vis_i = 2 visi=2 就跳过,打擂台,跑完全部后输出即可。
代码
#include<bits/stdc++.h>
using namespace std;const int N=2e5+5,M=5e3+5;
struct node{
int u,v,w;
};
int vis[M];
vector<node> G;int n,m,u,v,w;int fa[M];long long ans=0,cnt;
void init(){for(int i=1;i<=n;i++){fa[i]=i;}}
int find(int x) {
if (fa[x] != x) {
fa[x] = find(fa[x]);
}
return fa[x];
}
void merge(int x, int y) {
x = find(x);
y = find(y);
if (x != y) {
fa[y] = x;
}
}
bool cmp(node x,node y){return x.w<y.w;}
void kruskal(){sort(G.begin(),G.end(),cmp);
for(int i=0;i<m;i++){
int u=find(G[i].u),v=find(G[i].v);
if(u==v) continue;
ans+=G[i].w;
fa[u]=v;cnt++;vis[i]=1;
if(cnt==n-1) return ;
}
}void second_kruskal(){ans=0;
for(int i=0;i<m;i++){
if(vis[i]==2) continue;
int u=find(G[i].u),v=find(G[i].v);
if(u==v) continue;
ans+=G[i].w;
fa[u]=v;cnt++;
if(cnt==n-1) return ;
}
}
int main(){ios::sync_with_stdio(false);
cin.tie(0);cin>>n>>m;
init();
for(int i=1;i<=m;i++){
cin>>u>>v>>w;G.push_back((node){u,v,w});
}kruskal();long long maxans=1e18;;
for(int i=1;i<=m;i++){if(vis[i]==0) continue;
vis[i]=2;init();
second_kruskal();
maxans=min(maxans,ans);vis[i]=0;
}
cout<<maxans;
return 0;
}
如果要生成严格次小生成树,打擂台那加一个判断权值是否等于最小生成树的。