社区讨论

悬关,来个义父,一直49分,6 9-13没过

P4180[BJWC2010] 严格次小生成树参与者 1已保存回复 0

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
0 条
当前快照
1 份
快照标识符
@lyflsyp1
此快照首次捕获于
2024/07/10 16:54
2 年前
此快照最后确认于
2024/07/10 18:03
2 年前
查看原帖
CPP
#include<bits/stdc++.h>
#define int long long 
using namespace std;
const int N=600010;
int n,m,fa[N],v[N],t,f[N][20],g[N][20][2],d[N],ans,anss=0x7f;
int ver[N],head[N],Next[N],edge[N],tot;
struct node{
	int x,y,z;
}a[N];
bool cmp(node x,node y)
{
	return x.z<y.z;
}
int get(int x)
{
	if(fa[x]==x) return x;
	return fa[x]=get(fa[x]);
}
void add(int x,int y,int z)
{
	ver[++tot]=y,edge[tot]=z;
	Next[tot]=head[x],head[x]=tot;
}
void dfs(int x,int fa)
{
	for(int i=head[x];i;i=Next[i])
	{
		int y=ver[i],z=edge[i];
		if(y==fa) continue;
		f[y][0]=x;
		for(int j=1;j<=t;j++)
		f[y][j]=f[f[y][j-1]][j-1];
		d[y]=d[x]+1;
		dfs(y,x);
	}
}
void dfs2(int x,int fa)
{
	for(int i=head[x];i;i=Next[i])
	{
		int y=ver[i],z=edge[i];
		if(y==fa) continue;
		g[y][0][0]=z;
		for(int j=1;j<=t;j++)
		{
			g[y][j][0]=max(g[y][j-1][0],g[f[y][j-1]][j-1][0]);
			if(g[y][j-1][0]>g[f[y][j-1]][j-1][0])
			g[y][j][1]=max(g[y][j-1][1],g[f[y][j-1]][j-1][0]);
			if(g[y][j-1][0]<g[f[y][j-1]][j-1][0])
			g[y][j][1]=max(g[y][j-1][0],g[f[y][j-1]][j-1][1]);
			if(g[y][j-1][0]==g[f[y][j-1]][j-1][0])
			g[y][j][1]=max(g[y][j-1][1],g[f[y][j-1]][j-1][1]);
		}
		dfs2(y,x);
	}
}
signed main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=m;i++)
	scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].z);
	sort(a+1,a+m+1,cmp);
	for(int i=1;i<=n;i++) fa[i]=i;
	for(int i=1;i<=m;i++)
	{
		if(a[i].x==a[i].y) {v[i]=1; continue;}
		int x=get(a[i].x),y=get(a[i].y),z=a[i].z;
		if(x==y) continue;
		add(a[i].x,a[i].y,z),add(a[i].y,a[i].x,z);
		fa[x]=y;
		ans+=z;
		v[i]=1; 
	}
	t=log(n)/log(2)+1;
	d[1]=1;
	dfs(1,-1);
	dfs2(1,-1);
	for(int i=1;i<=m;i++)
	{
		if(v[i]) continue;
		int x=a[i].x,y=a[i].y,z=a[i].z,maxx=0,cmax=0;
		if(d[x]>d[y]) swap(x,y);
		for(int j=t;j>=0;j--)
		if(d[f[y][j]]>=d[x]) 
		{
			if(g[y][j][0]>maxx)
			{
				cmax=maxx;
				maxx=g[y][j][0];
			}
			else if(g[y][j][0]>cmax)
			cmax=g[y][j][0];
			y=f[y][j];
		}
		if(x==y)
		{
			if(z>maxx) anss=min(anss,z-maxx);
			if(z==maxx&&z>cmax) anss=min(anss,z-cmax);
			continue;
		}
		for(int j=t;j>=0;j--)
		{
			if(f[x][j]!=f[y][j]) 
			{
				if(g[x][j][0]>maxx)
			    {
				    cmax=maxx;
				    maxx=g[x][j][0];
			    }
			    else if(g[x][j][0]>cmax)
			    cmax=g[x][j][0];
			    
			    if(g[y][j][0]>maxx)
			    {
				    cmax=maxx;
				    maxx=g[y][j][0];
			    }
			    else if(g[y][j][0]>cmax)
			    cmax=g[y][j][0];
			    x=f[x][j],y=f[y][j];
			}
		}
		int u=g[x][0][0],v=g[y][0][0];
		if(u>maxx)
		{
			cmax=maxx;
			maxx=u;
		}
		else if(u>cmax&&u!=maxx)
		cmax=u;
		if(v>maxx)
		{
			cmax=maxx;
			maxx=v;
		}
		else if(v>cmax&&v!=maxx)
		cmax=v;
		
		if(z>maxx) anss=min(anss,z-maxx);
		if(z==maxx&&z>cmax) anss=min(anss,z-cmax);
	}
	printf("%d",ans+anss);
}

回复

0 条回复,欢迎继续交流。

正在加载回复...