BZOJ 1977 次小生成树 倍增LCA

11 篇文章 0 订阅
10 篇文章 0 订阅

题目大意:给定一个无向图,求严格次小生成树

这题纠结了我同学很久。。。首先如果是不严格的次小生成树(权值可以等于最小生成树),那么我们就有一个很简单明了的算法:

首先求出最小生成树 然后建立倍增LCA 枚举没进入最小生成树的边 在边的两端点的路径上寻找最大的边权 用最小生成树权值-最大边权+新边权去更新ans即可

但是这道题是严格次小生成树 那么也好办 我们记录一个次大边权 如果枚举到的边权和最大边权相等 就用路径上的次大边权代替最大边权去更新即可

于是我们只需要在倍增的时候记录最大权值和次大权值即可 注意建立倍增LCA的时候次大值的讨论 这里容易出问题

假期被罚十道题第一道。。。苦逼的娃,同学们都刷了四道了0.0 10%达成

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define M 100100
using namespace std;
typedef long long ll;
struct edge{
	int x,y,f;
	bool used; 
	bool operator < (const edge &y) const
	{
		return f < y.f;
	}
}edges[300300];
struct abcd{
	int to,f,next;
}table[M<<1];
int head[M],tot;
int n,m,fa[M][20],f_max[M][20],f_2th_max[M][20],dpt[M];
ll ans_min,ans_2th_min=~0ull>>1;
int belong[M];
int find(int x)
{
	if(!belong[x]||belong[x]==x) return belong[x]=x;
	return belong[x]=find(belong[x]);
}
void add(int x,int y,int z)
{
	table[++tot].to=y;
	table[tot].f=z;
	table[tot].next=head[x];
	head[x]=tot;
}
void dfs(int x)
{
	int i;
	dpt[x]=dpt[fa[x][0]]+1;
	for(i=head[x];i;i=table[i].next)
	{
		if(table[i].to==fa[x][0])
			continue;
		fa[table[i].to][0]=x;
		f_max[table[i].to][0]=table[i].f;
		dfs(table[i].to);
	}
}
int Find_F_Max(int x,int y)
{
	int j,re=0;
	if(dpt[x]<dpt[y])
		swap(x,y);
	for(j=19;~j;j--)
		if(dpt[fa[x][j]]>=dpt[y])
			re=max(re,f_max[x][j]),x=fa[x][j];
	if(x==y)
		return re;
	for(j=19;~j;j--)
		if(fa[x][j]!=fa[y][j])
		{
			re=max(re,f_max[x][j]);
			re=max(re,f_max[y][j]);
			x=fa[x][j];
			y=fa[y][j];
		}
	re=max(re,f_max[x][0]);
	re=max(re,f_max[y][0]);
	return re;
}
int Find_F_2th_Max(int x,int y,int z)
{
	int j,re=0;
	if(dpt[x]<dpt[y])
		swap(x,y);
	for(j=19;~j;j--)
		if(dpt[fa[x][j]]>=dpt[y])
		{
			if(f_max[x][j]!=z)
				re=max(re,f_max[x][j]);
			else
				re=max(re,f_2th_max[x][j]);
			x=fa[x][j];
		}
	if(x==y)
		return re;
	for(j=19;~j;j--)
		if(fa[x][j]!=fa[y][j])
		{
			if(f_max[x][j]!=z)
				re=max(re,f_max[x][j]);
			else
				re=max(re,f_2th_max[x][j]);
			if(f_max[y][j]!=z)
				re=max(re,f_max[y][j]);
			else
				re=max(re,f_2th_max[y][j]);
			x=fa[x][j];
			y=fa[y][j];
		}
	j=0;
	if(f_max[x][j]!=z)
		re=max(re,f_max[x][j]);
	else
		re=max(re,f_2th_max[x][j]);
	if(f_max[y][j]!=z)
		re=max(re,f_max[y][j]);
	else
		re=max(re,f_2th_max[y][j]);
	return re;
}
int main()
{
	int i,j;
	cin>>n>>m;
	for(i=1;i<=m;i++)
		scanf("%d%d%d",&edges[i].x,&edges[i].y,&edges[i].f);
	sort(edges+1,edges+m+1);
	for(i=1;i<=m;i++)
	{
		int fx=find(edges[i].x),fy=find(edges[i].y);
		if(fx!=fy)
		{
			edges[i].used=true;
			belong[fx]=fy;
			ans_min+=edges[i].f;
			add(edges[i].x,edges[i].y,edges[i].f);
			add(edges[i].y,edges[i].x,edges[i].f);
		}
	}
	dfs(1);
	for(j=1;j<=19;j++)
		for(i=1;i<=n;i++)
		{
			int f=fa[i][j-1];
			fa[i][j]=fa[f][j-1];
			if( f_max[i][j-1] > f_max[f][j-1] )
			{
				f_max[i][j]=f_max[i][j-1];
				f_2th_max[i][j]=max( f_2th_max[i][j-1] , f_max[f][j-1] );
			}
			else
			{
				f_max[i][j]=f_max[f][j-1];
				if( f_max[i][j-1] < f_max[f][j-1] )
					f_2th_max[i][j]=max( f_2th_max[f][j-1] , f_max[i][j-1] );
				else
					f_2th_max[i][j]=max(  f_2th_max[i][j-1] , f_2th_max[f][j-1] );
			}
			f_max[i][j]=max( f_max[i][j-1] , f_max[f][j-1] );
		}
	for(i=1;i<=m;i++)
		if(!edges[i].used)
		{
			int temp=Find_F_Max(edges[i].x,edges[i].y);
			if(temp==edges[i].f)
				temp=Find_F_2th_Max(edges[i].x,edges[i].y,temp);
			ans_2th_min=min(ans_2th_min,ans_min-temp+edges[i].f);
		}
	cout<<ans_2th_min<<endl;
}



  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
题目描述 有一个 $n$ 个点的棋盘,每个点上有一个数字 $a_i$,你需要从 $(1,1)$ 走到 $(n,n)$,每次只能往右或往下走,每个格子只能经过一次,路径上的数字和为 $S$。定义一个点 $(x,y)$ 的权值为 $a_x+a_y$,所有满足条件的路径中,所有点的权值和的最小值。 输入格式 第一行一个整数 $n$。 接下来 $n$ 行,每行 $n$ 个整数,表示棋盘上每个点的数字。 输出格式 输出一个整数,表示所有满足条件的路径中,所有点的权值和的最小值。 数据范围 $1\leq n\leq 300$ 输入样例 3 1 2 3 4 5 6 7 8 9 输出样例 25 算法1 (树形dp) $O(n^3)$ 我们可以先将所有点的权值出来,然后将其看作是一个有权值的图,问题就转化为了在这个图中从 $(1,1)$ 到 $(n,n)$ 的所有路径中,所有点的权值和的最小值。 我们可以使用树形dp来解决这个问题,具体来说,我们可以将这个图看作是一棵树,每个点的父节点是它的前驱或者后继,然后我们从根节点开始,依次向下遍历,对于每个节点,我们可以考虑它的两个儿子,如果它的两个儿子都被遍历过了,那么我们就可以计算出从它的左儿子到它的右儿子的路径中,所有点的权值和的最小值,然后再将这个值加上当前节点的权值,就可以得到从根节点到当前节点的路径中,所有点的权值和的最小值。 时间复杂度 树形dp的时间复杂度是 $O(n^3)$。 C++ 代码 算法2 (动态规划) $O(n^3)$ 我们可以使用动态规划来解决这个问题,具体来说,我们可以定义 $f(i,j,s)$ 表示从 $(1,1)$ 到 $(i,j)$ 的所有路径中,所有点的权值和为 $s$ 的最小值,那么我们就可以得到如下的状态转移方程: $$ f(i,j,s)=\min\{f(i-1,j,s-a_{i,j}),f(i,j-1,s-a_{i,j})\} $$ 其中 $a_{i,j}$ 表示点 $(i,j)$ 的权值。 时间复杂度 动态规划的时间复杂度是 $O(n^3)$。 C++ 代码
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值