P4180-[BJWC2010]严格次小生成树【Kruskal,倍增】

1|0正题

题目链接:https://www.luogu.com.cn/problem/P4180

---

1|1题目大意

$n$个点$m$条边的一张无向图，求它的严格次小生成树。

$1\leq n\leq 10^5,1\leq m\leq 3\times 10^5$

---

1|2解题思路

一定存在一种严格次小生成树和最小生成树只差一条边，感性理解的话大概就是如果有两条不同那么肯定有一条可以替换成另一条要么更优要么不变。

所以我们可以枚举一条不选的边$(u,v,w)$然后找到最小生成树上$u,v$路径最大的权值$k$替换。

但是发现有可能恰好$w=k$，所以我们不只需要统计最大的权值，还要统计一个严格次大的，如果等于就选择严格次大的。

时间复杂度$O(n\log n)$

---

1|3code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const ll N=1e5+10,T=18;
struct enode{
	ll x,y,w,v;
}e[3*N];
struct node{
	ll to,next,w;
}a[N<<1];
ll n,m,tot,ls[N],fa[N],dep[N];
ll ans,sum,f[N][T],g[N][T][2];
bool cmp(enode x,enode y)
{return x.w<y.w;}
ll find(ll x)
{return (fa[x]==x)?x:(fa[x]=find(fa[x]));}
void addl(ll x,ll y,ll w){
	a[++tot].to=y;
	a[tot].next=ls[x];
	ls[x]=tot;a[tot].w=w;
	return;
}
void dfs(ll x,ll fa){
	dep[x]=dep[fa]+1;
	for(ll i=ls[x];i;i=a[i].next){
		ll y=a[i].to;
		if(y==fa)continue;
		f[y][0]=x;g[y][0][0]=a[i].w;
		dfs(y,x);
	}
	return;
}
void calc(ll &mx,ll &mi,ll x){
	if(x>mx)mi=mx,mx=x;
	else if(x>mi&&x!=mx)mi=x;
	return;
}
void calccc(ll x,ll y,ll w){
	if(x==y)return;
	if(dep[x]>dep[y])swap(x,y);
	ll mx=-1,mi=-1;
	for(ll i=T-1;i>=0;i--)
		if(dep[f[y][i]]>=dep[x]){
			calc(mx,mi,g[y][i][0]);
			calc(mx,mi,g[y][i][1]);
			y=f[y][i];
		}
	if(x!=y){
		for(ll i=T-1;i>=0;i--)
			if(f[x][i]!=f[y][i]){
				calc(mx,mi,g[x][i][0]);
				calc(mx,mi,g[x][i][1]);
				calc(mx,mi,g[y][i][0]);
				calc(mx,mi,g[y][i][1]);
				x=f[x][i];y=f[y][i];
			}
		calc(mx,mi,g[x][0][0]);
		calc(mx,mi,g[x][0][1]);
		calc(mx,mi,g[y][0][0]);
		calc(mx,mi,g[y][0][1]);
	}
	if(w!=mx)ans=min(ans,sum+w-mx);
	else if(mi>=0)ans=min(ans,sum+w-mi);
	return;
}
signed main()
{
	scanf("%lld%lld",&n,&m);
	for(ll i=1;i<=m;i++)
		scanf("%lld%lld%lld",&e[i].x,&e[i].y,&e[i].w);
	sort(e+1,e+1+m,cmp);
	for(ll i=1;i<=n;i++)fa[i]=i;
	for(ll i=1;i<=m;i++){
		ll x=find(e[i].x),y=find(e[i].y);
		if(x==y)continue;
		addl(e[i].x,e[i].y,e[i].w);
		addl(e[i].y,e[i].x,e[i].w);
		fa[x]=y;e[i].v=1;sum+=e[i].w;
	}
	memset(g,-1,sizeof(g));
	dfs(1,1);
	for(ll j=1;j<T;j++)
		for(ll i=1;i<=n;i++){
			f[i][j]=f[f[i][j-1]][j-1];
			calc(g[i][j][0],g[i][j][1],g[i][j-1][0]);
			calc(g[i][j][0],g[i][j][1],g[i][j-1][1]);
			calc(g[i][j][0],g[i][j][1],g[f[i][j-1]][j-1][0]);
			calc(g[i][j][0],g[i][j][1],g[f[i][j-1]][j-1][1]);
		}
	ans=1e18;
	for(ll i=1;i<=m;i++)
		if(!e[i].v)calccc(e[i].x,e[i].y,e[i].w);
	printf("%lld\n",ans);
	return 0;
}

__EOF__

本文作者：QuantAsk
本文链接：https://www.cnblogs.com/QuantAsk/p/15021981.html
关于博主：退役OIer，GD划水选手
版权声明：本博客所有文章除特别声明外，均采用 BY-NC-SA 许可协议。转载请注明出处！
声援博主：如果您觉得文章对您有帮助，可以点击文章右下角【推荐】一下。您的鼓励是博主的最大动力！