次小生成树

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define max(a,b) a>b?a:b
const int N=100010;
int n,m,sum,tot,head[N],f[N][20],dep[N];
int ans=1e18,fa[N],colcnt,mx[N][20],se[N][20];
struct edge{int to,next,w;}e[N<<1];
struct EDGE{int u,v,w,flag;}ee[N<<1];
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;
}
void add_edge(int u,int v,int w)
{
	e[++tot].to=v;
	e[tot].w=w;
	e[tot].next=head[u];
	head[u]=tot;
}
bool cmp(EDGE aa,EDGE bb){return aa.w<bb.w;}
int find(int x)
{
	if(fa[x]==x) return x;
	return fa[x]=find(fa[x]);
}
void dfs_init(int u)
{
	for(int i=head[u];i;i=e[i].next)
	{
		int v=e[i].to;
		if(v==f[u][0]) continue;
		dep[v]=dep[u]+1;
		mx[v][0]=e[i].w;
		f[v][0]=u;
		dfs_init(v);
	}
}
int LCA2(int u,int v)
{
	int flag;
	if(dep[u]<dep[v]) swap(u,v);
	for(int i=0;i<=18;i++)
	{
		if((1<<i)>dep[u])
		{
			flag=i-1;
			break;
		}
	}
	for(int i=flag;i>=0;i--) if(dep[u]>=dep[v]+(1<<i)) u=f[u][i];
	if(u==v) return u;
	for(int i=18;i>=0;i--)
	{
		if(f[u][i]!=f[v][i])
		{
			u=f[u][i];
			v=f[v][i];
		}
	}
	return f[u][0];
}
int search(int u,int v,int w)
{
	int lca=LCA2(u,v),flag,lmax,rmax;
	lmax=rmax=0;
	for(int i=0;i<=18;i++)
	{
		if((1<<i)>dep[u])
		{
			flag=i-1;
			break;
		}
	}
	for(int i=flag;i>=0;i--)
	{
		if(dep[u]>dep[lca]+(1<<i))
		{
			if(mx[u][i]!=w) lmax=max(lmax,mx[u][i]);
			else lmax=max(lmax,se[u][i]);
			u=f[u][i];
		}
	}
	for(int i=0;i<=18;i++)
	{
		if((1<<i)>dep[v])
		{
			flag=i-1;
			break;
		}
	}
	for(int i=flag;i>=0;i--)
	{
		if(dep[v]>=dep[lca]+(1<<i))
		{
			if(mx[v][i]!=w) rmax=max(rmax,mx[v][i]);
			else rmax=max(rmax,se[v][i]);
		}
		v=f[v][i];
	}
	return max(lmax,rmax);
}
void LCA1()
{
	for(int i=1;i<=18;i++)
	for(int j=1;j<=n;j++)
	{
		f[j][i]=f[f[j][i-1]][i-1];
		mx[j][i]=max(mx[j][i-1],mx[f[j][i-1]][i-1]);
		se[j][i]=max(se[j][i-1],se[f[j][i-1]][i-1]);
		if(mx[j][i-1]<mx[f[j][i-1]][i-1]&&se[j][i]<mx[j][i-1])
			se[j][i]=mx[j][i-1];
			
		else if(mx[j][i-1]>mx[f[j][i-1]][i-1]
		      &&se[j][i]<mx[f[j][i-1]][i-1])
			se[j][i]=mx[f[j][i-1]][i-1];
	}
}
signed main()
{
	n=read(),m=read();
	for(int i=1;i<=n;i++) fa[i]=i;
	for(int i=1;i<=m;i++)
	{
		ee[i].u=read();
		ee[i].v=read();
		ee[i].w=read();
		ee[i].flag=0;
	}
	sort(ee+1,ee+m+1,cmp);
	for(int i=1;i<=m;i++)
	{
		int uu=find(ee[i].u);
		int vv=find(ee[i].v);
		if(uu==vv) continue;
		colcnt++;
		sum+=ee[i].w;
		ee[i].flag=1;
		fa[vv]=uu;
		add_edge(ee[i].u,ee[i].v,ee[i].w);
		add_edge(ee[i].v,ee[i].u,ee[i].w);
		if(colcnt==n-1) break;
	}
	dfs_init(1);
	LCA1();
	for(int i=1;i<=m;i++)
	{
		if(ee[i].flag) continue;
		int tmp=search(ee[i].u,ee[i].v,ee[i].w);
		if(tmp!=ee[i].w&&ans>sum-tmp+ee[i].w)
			ans=sum-tmp+ee[i].w;
	}
	printf("%lld",ans);
	return 0;
}

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

xiyuping24

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值