题目描述
小C最近学了很多最小生成树的算法,Prim算法、Kurskal算法、消圈算法等等。正当小C洋洋得意之时,小P又来泼小C冷水了。小P说,让小C求出一个无向图的次小生成树,而且这个次小生成树还得是严格次小的,也就是说:如果最小生成树选择的边集是EM,严格次小生成树选择的边集是ES,那么需要满足:和严格小于
这下小 C 蒙了,他找到了你,希望你帮他解决这个问题。
输入格式
第一行包含两个整数N 和M,表示无向图的点数与边数。 接下来 M行,每行 3个数x y z 表示,点 x 和点y之间有一条边,边的权值为z。
输出格式
包含一行,仅一个数,表示严格次小生成树的边权和。(数据保证必定存在严格次小生成树)
题解:由kruskal算法,我们不难想到,在剩下没有选择的边中,我们选一条把它和在树中的图换一下,然后判断这棵新树的大小,还要存一下次小边,因为如果当前加入的边的权值与最大值相等,就不满足严格次小,所以再搞一下次小边,就阔以了
(说是好说,写可挺麻烦,我调了一上午,最后还是吸氧水过去的嘤嘤嘤 )
上代码
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#define INF 2147483647000000
#define int long long
using namespace std;
const int M=600004;
const int N=200002;
int n,m,tot,head[M],nex[M],to[M],v[M];
int dep[N];
int w[N][31],f[N][31],minn[N][31];
bool vis[M];
int ans,cnt;
struct NODE{
int x,y,z;
bool operator < (const NODE &a){
return this->z <a.z;
}
}bian[M];
int fa[N];
inline int read(){
int x=0,f=1;
char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
int find(int x) {//并查集找父亲
if(fa[x]!=x) fa[x]=find(fa[x]);
return fa[x];
}
inline void add2(int x,int y,int z){
++tot;
nex[tot]=head[x];
head[x]=tot;
to[tot]=y;
v[tot]=z;
}
inline void kruskal(){
ans=0,cnt=0;
for(int i=1;i<=m;i++){
if(fa[find(bian[i].x)]!=fa[find(bian[i].y)]){
fa[find(bian[i].x)]=fa[find(bian[i].y)];
cnt++;
vis[i]=1;
ans+=bian[i].z;
add2(bian[i].x,bian[i].y,bian[i].z);
add2(bian[i].y,bian[i].x,bian[i].z);
}
if(cnt==n-1) break;
}
return ;
}
void dfs(int x,int fath){
for(int i=head[x];i;i=nex[i]){
if(to[i]!=fath){
dep[to[i]]=dep[x]+1;
w[to[i]][0]=v[i];
f[to[i]][0]=x;
minn[to[i]][0]=-INF;
dfs(to[i],x);
}
}
}
inline int lca(int x,int y){
if(dep[x]>dep[y]) swap(x,y);
for(int i=30;i>=0;i--) if(dep[x]<=dep[y]-(1<<i)) y=f[y][i];
if(x==y) return x;
for(int i=30;i>=0;i--) {
if(f[x][i]!=f[y][i]){
x=f[x][i];
y=f[y][i];
}
}
return f[x][0];
}
int qsum(int x,int y,int maxx){
int maxxx=-INF;
for(int i=30;i>=0;i--){
if(dep[f[x][i]]>=dep[y]){
if(maxx!=w[x][i]) maxxx=max(maxxx,w[x][i]);
else maxxx=max(maxxx,minn[x][i]);
}
}
return maxxx;
}
main(){
n=read();m=read();
for(int i=1;i<=m;i++){
bian[i].x=read();bian[i].y=read();bian[i].z=read();
}
sort(bian+1,bian+m+1);
for(int i=1;i<=n;i++) fa[i]=i;
kruskal();
dep[1]=1;
f[1][0]=0;
minn[1][0]=-INF;
memset(w,0,sizeof(w));
dfs(1,1);
for(int i=1;i<=30;i++){
for(int j=1;j<=n;j++){
f[j][i]=f[f[j][i-1]][i-1];
w[j][i]=max(w[j][i-1],w[f[j][i-1]][i-1]);
minn[j][i]=max(minn[j][i-1],minn[f[j][i-1]][i-1]);
if(w[j][i-1]>w[f[j][i-1]][i-1]) minn[j][i]=max(minn[j][i],w[f[j][i-1]][i-1]);
else if(w[j][i-1]<w[f[j][i-1]][i-1]) minn[j][i]=max(minn[j][i],w[j][i-1]);
}
}
int Ans=INF;
for(int i=1;i<=m;i++){
if(!vis[i]){
int x=bian[i].x;
int y=bian[i].y;
int k=lca(x,y);
int x1=qsum(x,k,bian[i].z);
int x2=qsum(y,k,bian[i].z);
Ans=min(Ans,ans-max(x1,x2)+bian[i].z);
}
}
printf("%lld",Ans);
return 0;
}