[BJWC2010]严格次小生成树

板子题

先kruskal得到最小生成树,再枚举未加进生成树中的边i,该边i和生成树树形成了一个环,选择这个环上最大的不等于i的权值的边去掉就得到一个新的树,所有这样得到的树权值之和即为次小生成树。

求环上最大边可以用倍增或者树链剖分,由于最大边可能和才加上去的边相等,所以还要维护一个次大边

#include<bits/stdc++.h>
#define N 100005
#define M 300005
using namespace std;
typedef long long ll;
const ll INF = 10000000000000000;
int n,m,fa[N],f[N][20],dep[N];
bool vis[M];
ll maxx[N][20],naxx[N][20],ans,minn;
struct E {int u,v;ll w;} e[M];
struct Edge {int next,to;ll dis;}edge[M<<1];
int head[N],cnt=1;
void add_edge(int from,int to,ll dis)
{
	edge[++cnt].next=head[from];
	edge[cnt].to=to;
	edge[cnt].dis=dis;
	head[from]=cnt;
}
template <class T>
void read(T &x)
{
	char c;int sign=1;
	while((c=getchar())>'9'||c<'0') if(c=='-') sign=-1; x=c-48;
	while((c=getchar())>='0'&&c<='9') x=(x<<1)+(x<<3)+c-48; x*=sign;
}
bool cmp(E a,E b) {return a.w<b.w;}
int find(int x) {return x==fa[x] ? x : fa[x]=find(fa[x]);}
void kruskal()
{
	sort(e+1,e+m+1,cmp);
	int cnt=0; minn=0;
	for(int i=1;i<=m;++i)
	{
		int fu=find(e[i].u),fv=find(e[i].v);
		if(fu!=fv)
		{
			fa[fu]=fv;
			add_edge(e[i].u,e[i].v,e[i].w);
			add_edge(e[i].v,e[i].u,e[i].w);
			vis[i]=1;
			minn+=e[i].w;
			if(++cnt==n-1) break;
		}
	}
}
void dfs(int rt,int fa)
{
	dep[rt]=dep[fa]+1;
	for(int i=head[rt];i;i=edge[i].next)
	{
		int v=edge[i].to;
		if(v==fa) continue;
		maxx[v][0]=edge[i].dis;
		naxx[v][0]=0;
		f[v][0]=rt;
		for(int i=1;i<20;++i)
		{
			f[v][i]=f[f[v][i-1]][i-1];
			maxx[v][i]=max(maxx[v][i-1],maxx[f[v][i-1]][i-1]);
			naxx[v][i]=max(naxx[v][i-1],naxx[f[v][i-1]][i-1]);
			if(maxx[v][i-1]!=maxx[v][i]) naxx[v][i]=max(naxx[v][i],maxx[v][i-1]);
			if(maxx[f[v][i-1]][i-1]!=maxx[v][i]) naxx[v][i]=max(naxx[v][i],maxx[f[v][i-1]][i-1]);
		}
		dfs(v,rt);
	}
}
ll query(int x,int y,ll MAX)
{
	ll ret=-INF;
	if(dep[x]<dep[y]) swap(x,y);
	for(int i=19;i>=0;--i)
	{
		if(dep[f[x][i]]>=dep[y])
		{
			if(maxx[x][i]!=MAX) ret=max(ret,maxx[x][i]);
			else ret=max(ret,naxx[x][i]);
			x=f[x][i];
		}
	}
	if(x==y) return ret;
	for(int i=19;i>=0;--i)
	{
		if(f[x][i]!=f[y][i])
		{
			if(maxx[x][i]!=MAX) ret=max(ret,maxx[x][i]);
			else ret=max(ret,naxx[x][i]);
			if(maxx[y][i]!=MAX) ret=max(ret,maxx[y][i]);
			else ret=max(ret,naxx[y][i]);
			x=f[x][i];y=f[y][i];
		}
	}
	if(maxx[x][0]!=MAX) ret=max(ret,maxx[x][0]); else ret=max(ret,naxx[x][0]);
	if(maxx[y][0]!=MAX) ret=max(ret,maxx[y][0]); else ret=max(ret,naxx[y][0]);
	return ret;
}
int main()
{
	read(n);read(m);
	for(int i=1;i<=n;++i) fa[i]=i;
	for(int i=1;i<=m;++i) read(e[i].u),read(e[i].v),read(e[i].w);
	kruskal();
	ans=INF;
	dfs(1,0);
	for(int i=1;i<=m;++i)
	{
		if(vis[i]) continue;
		ll q=query(e[i].u,e[i].v,e[i].w);
		if(minn-q+e[i].w!=minn) ans=min(ans,minn-q+e[i].w);
	}
	cout<<ans<<endl;
	return 0;
}
原文地址:https://www.cnblogs.com/Chtholly/p/11170161.html