P4180 [BJWC2010]严格次小生成树

题目

题目

思路

最小生成树都会吧?
不会的戳这里
接下来我们用LCA求每一条非树边在树上环的最边权和次大边权,然后求一个min(s-mx+v,s-mx2+v,mn)就可以了,注意如果mx=v,第一项不比较.
code:

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<cmath>
using namespace std;
long long fa[100010][22],v[100010][22][2],head[500010],c[500010],x,y,f[100010];
long long n,m,s;
struct f{
    
    
	long long to,net,w;
} a[600020];
struct PVZ{
    
    
	long long x,y,z;
	bool o;
} z[300010];
long long k;
bool cmp(PVZ x,PVZ y)
{
    
    
	return x.z<y.z;
}
long long find(long long x)
{
    
    
	if (f[x]==x) return x;
	else return f[x]=find(f[x]);
}
void add(long long u,long long v,long long z)
{
    
    
    a[k].to=v;
    a[k].w=z;
	a[k].net=head[u];
    head[u]=k++;
    return; 
}
void dk()
{
    
    
	for (long long j=1;j<=m;j++)
	{
    
    
		if (find(z[j].x)!=find(z[j].y))
		{
    
    
			z[j].o=1;
			f[find(z[j].x)]=find(z[j].y);
			s+=z[j].z;
			add(z[j].x,z[j].y,z[j].z);
			add(z[j].y,z[j].x,z[j].z);
		}
	}
	return;
}
void dfs(long long x,long long f,long long g)
{
    
    
	fa[x][0]=f;
	v[x][0][0]=g;
	v[x][0][1]=-2147483647000000;
	c[x]=c[f]+1;
	for (long long j=1;(1<<j)<=c[x];j++)
	{
    
    
		fa[x][j]=fa[fa[x][j-1]][j-1];
		v[x][j][0]=max(v[x][j-1][0],v[fa[x][j-1]][j-1][0]);
		v[x][j][1]=max(v[x][j-1][1],v[fa[x][j-1]][j-1][1]);
		if (v[x][j][0]>v[x][j-1][0]) v[x][j][1]=max(v[x][j][1],v[x][j-1][0]);
		if (v[x][j][0]>v[fa[x][j-1]][j-1][0]) v[x][j][1]=max(v[x][j][1],v[fa[x][j-1]][j-1][0]);
	}
	for (long long i=head[x];i!=-1;i=a[i].net)
	{
    
    
		if (a[i].to!=f) dfs(a[i].to,x,a[i].w);
	}
	return;
}
long long mn=2147483647000000;
long long LCA(long long x,long long y,long long z)
{
    
    
	if (c[x]>c[y]) swap(x,y);
	long long mx1=0,mx2=0;
	for (long long i=20;i>=0;i--)
	{
    
    
		if (c[x]<=c[y]-(1<<i))
		{
    
    
			mx1=max(mx1,v[y][i][0]);
			mx2=max(mx2,v[y][i][1]);
			if (mx1!=v[y][i][0]) mx2=max(mx2,v[y][i][0]);
			y=fa[y][i];
		}
	}
	if (x!=y)
	{
    
    
		for (long long i=20;i>=0;i--)
		{
    
    
			if (fa[x][i]==fa[y][i]) continue;
			else
			{
    
    
				mx1=max(mx1,v[y][i][0]);
				mx2=max(mx2,v[y][i][1]);
				if (mx1!=v[y][i][0]) mx2=max(mx2,v[y][i][0]);
				mx1=max(mx1,v[x][i][0]);
				mx2=max(mx2,v[x][i][1]);
				if (mx1!=v[x][i][0]) mx2=max(mx2,v[x][i][0]);
				x=fa[x][i],y=fa[y][i];
			}
		}
	}
	mx1=max(mx1,v[x][0][0]);
	mx2=max(mx2,v[x][0][1]);
	if (mx1!=v[x][0][0]) mx2=max(mx2,v[x][0][0]);
	mx1=max(mx1,v[y][0][0]);
	mx2=max(mx2,v[y][0][1]);
	if (mx1!=v[y][0][0]) mx2=max(mx2,v[y][0][0]);
	if (z==mx1) mn=min(mn,s-mx2+z);
	if (z>mx1) mn=min(mn,s-mx1+z);
	return (x!=y?fa[x][0]:x);
}
int main()
{
    
    
    memset(head,-1,sizeof(head));
	cin>>n>>m;
	for (long long i=1;i<=n;i++) f[i]=i;
	for (long long i=1;i<=m;i++)
	{
    
    
		cin>>z[i].x>>z[i].y>>z[i].z;
		if (z[i].x==z[i].y)
		{
    
    
			i--,m--;
		}
	}
	sort(z+1,z+1+m,cmp);
	dk();
	dfs(1,0,0);
	for (long long i=1;i<=m;i++)
	{
    
    
		if (z[i].o||(z[i].x==z[i].y)) continue;
		LCA(z[i].x,z[i].y,z[i].z);
	}
	cout<<mn;
    return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_49843717/article/details/115021279