树和森林

最新推荐文章于 2023-01-20 22:17:12 发布
最新推荐文章于 2023-01-20 22:17:12 发布
阅读量164 收藏 1
点赞数 1
分类专栏： ------动态规划------ # 树形dp # 树
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/Tan_tan_tann/article/details/109430197
版权

树和森林

题解

很好骗分的一道题，有人就在场上骗到了90pts，例子就是我。

我们可以先来看一看部分分。

14pts:我们直接枚举两个连通块上相连的点，O\left(n^2 \right )的算法。

20pts:很容易发现，在两个联通块上互相相连的点是谁并不重要，记一个点到块上每个点的路径长度总和为f_{u}，记一个联通块中最大的f值为mxf_{i}，答案就是sz_{1}sz_{2}+sz_{1}mxf_{2}+sz_{2}mxf_{1}+sumf_{1}+sumf_{2}，其中sumf表示这个连通块自身的贡献，f值可以用一个换根dp来求出，时间复杂度O\left(n \right )。

30pts:再将三个联通块中的中间的连通块与其它连通块相连接的点枚举，答案为\tiny (mxf_{a}+sz_{a})(n-sz_{a})+(mxf_{c}+sz_{c})(n-sz_{c})+d_{b1,b2}sz_{a}sz_{c}+f_{b1}sz_{a}+f_{b2}sz_{c}+sumf_{a}+sumf_{b}+sumf_{c}

其中点b为中间块。由于d_{b1,b2}可以通过倍增求出，时间复杂度O\left(n^2log_{n})。

70pts:对于子任务2，我们可以通过树形dp快速的求出，只用判断该节点与其子节点是否相连即可，分别存储该节点与其儿子连边为奇数或是偶数，跑一遍dfs。时间复杂度O\left(n \right )。

90pts:容易证明，对于一棵树，要么其无法满足拆分条件，要么有唯一解。因为叶节点的奇偶性是固定的，故其父亲节点的来连边状况也是固定的，若根节点符合条件则有唯一解，否则无解，用70pts的做法可以求出。于是笔者就得到了90pts。

100pts:我们发现，在30pts的做法中，只有d_{b1,b2}sz_{a}sz_{c}+f_{b1}sz_{a}+f_{b2}sz_{c}一项不是固定的，其它都可以预处理出来，我们只用考虑如何求出最大的d_{b1,b2}sz_{a}sz_{b}+f_{b1}sz_{a}+f_{b1}sz_{c}。我们可以将d_{b1,b2}拆成dep_{b1}+dep_{b2}-2dep_{lca(b1,b2)}。于是，原式就成了sz_{a}(dep_{b1}sz_{c}+f_{b1})+sz_{c}(dep_{b2}sz_{a}+f_{b2})-2dep_{lca(b1,b2)}sz_{a}sz_{c}。我们可以在dfs的过程中，对于每个节点求出它作为lca时上式的最大值，可以通过树形dp来维护。

源码

笔者代码有些冗长。

#include<cstdio>
#include<cmath>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include<map>
#include<set>
using namespace std;
#define MAXN 100005
typedef long long LL;
typedef unsigned long long uLL;
const int INF=0x7f7f7f7f;
typedef pair<int,int> pii;
template<typename _T>
_T Fabs(_T x){return x<0?-x:x;}
template<typename _T>
void read(_T &x){
	_T f=1;x=0;char s=getchar();
	while(s>'9'||s<'0'){if(s=='-')f=-1;s=getchar();}
	while('0'<=s&&s<='9'){x=(x<<3)+(x<<1)+(s^48);s=getchar();}
	x*=f;
}
int n,m,rt[5],sz[5],dep[MAXN];
LL mf[MAXN],sumf[MAXN],f[MAXN],g[MAXN],siz[MAXN];
LL mx1[MAXN],mx2[MAXN],mx;
int dp[MAXN][2],fi[MAXN],sta[MAXN],stak,ff[MAXN][20];
bool col[MAXN],dw[MAXN][2];char maze[MAXN];
int belong[MAXN],cnt,head[MAXN],tot;
struct edge{int to,nxt,id;}e[MAXN<<1];
void addEdge(int u,int v,int i){e[++tot]=(edge){v,head[u],i};head[u]=tot;}
void dfs1(int u,int fa){
	g[u]=0;sz[belong[u]]++;siz[u]=1;dep[u]=dep[fa]+1;
	ff[u][0]=fa;for(int i=1;i<19;i++)ff[u][i]=ff[ff[u][i-1]][i-1];
	for(int i=head[u];i;i=e[i].nxt){
		int v=e[i].to;if(v==fa)continue;
		belong[v]=cnt;dfs1(v,u);
		siz[u]+=siz[v];g[u]+=g[v]+siz[v];
	}
}
void dfs2(int u,int fa){
	f[u]=g[u];
	for(int i=head[u];i;i=e[i].nxt){
		int v=e[i].to;if(v==fa)continue;
		g[u]-=g[v]+siz[v];siz[u]-=siz[v];
		g[v]+=g[u]+siz[u];siz[v]+=siz[u];
		dfs2(v,u);
		siz[v]-=siz[u];g[v]-=g[u]+siz[u];
		siz[u]+=siz[v];g[u]+=g[v]+siz[v];
	}
}
void dfs3(int u,int fa){
	int res=0,md=INF,sum=0,mdd=-1;
	if(!head[u]){dp[u][0]=0;dp[u][1]=-INF;return ;}
	for(int i=head[u];i;i=e[i].nxt){
		int v=e[i].to;if(v==fa)continue;
		fi[v]=e[i].id;dfs3(v,u);
		if(col[v]==1){
			res+=max(dp[v][1],dp[v][0]+1);
			if(dp[v][0]+1>=dp[v][1])sum++;
			if(dp[v][1]>=0&&dp[v][0]>=0){
				if(Fabs(dp[v][1]-dp[v][0]-1)<md);
					md=Fabs(dp[v][1]-dp[v][0]-1),mdd=v;
			}
		}
		else{
			res+=max(dp[v][0],dp[v][1]+1);
			if(dp[v][1]+1>=dp[v][0])sum++;
			if(dp[v][0]>=0&&dp[v][1]>=0){
				if(Fabs(dp[v][1]-dp[v][0]-1)<md)
					md=Fabs(dp[v][0]-dp[v][1]-1),mdd=v;
			}
		}
	}
	if(sum&1)dp[u][1]=res,dp[u][0]=(md<=res?res-md:-INF);
	else dp[u][0]=res,dp[u][1]=(md<=res?res-md:-INF);
	for(int i=head[u];i;i=e[i].nxt){
		int v=e[i].to;if(v==fa)continue;
		if(col[v]==1){
			if(dp[v][0]+1>=dp[v][1]){
				dw[v][sum&1]=1;
				if(mdd!=v)dw[v][(sum&1)^1]=1;
			}
			else if(mdd==v)dw[v][(sum&1)^1]=1;
		}
		else{
			if(dp[v][1]+1>=dp[v][0]){
				dw[v][sum&1]=1;
				if(mdd!=v)dw[v][(sum&1)^1]=1;
			}
			else if(mdd==v)dw[v][(sum&1)^1]=1;
		}
	}
}
int lca(int a,int b){
	if(dep[a]>dep[b])swap(a,b);
	for(int i=18;i>=0;i--)if(dep[ff[b][i]]>=dep[a])b=ff[b][i];
	if(a==b)return a;
	for(int i=18;i>=0;i--)if(ff[a][i]!=ff[b][i])a=ff[a][i],b=ff[b][i];
	return ff[a][0];
}
void dfs4(int u,int fa,int s){
	for(int i=head[u];i;i=e[i].nxt){
		int v=e[i].to;if(v==fa)continue;
		if(dw[v][s])sta[++stak]=fi[v],dfs4(v,u,col[v]^1);
		else dfs4(v,u,col[v]);
	}
}
void dfs5(int u,int fa,int a,int b){
	mx1[u]=f[u]*a;mx2[u]=f[u]*b;mx=max(mx,mx1[u]+mx2[u]);
	for(int i=head[u];i;i=e[i].nxt){
		int v=e[i].to;if(v==fa)continue;
		dfs5(v,u,a,b);LL c=1ll*a*b;
		mx=max(mx,mx1[u]+mx2[v]+c);
		mx=max(mx,mx2[u]+mx1[v]+c);
		mx1[u]=max(mx1[u],mx1[v]+c);
		mx2[u]=max(mx2[u],mx2[v]+c);
	}
}
LL getmx(int rt,int a,int b){mx=0;dfs5(rt,0,a,b);return mx;}
signed main(){
	//freopen("lct.in","r",stdin);
	//freopen("lct.out","w",stdout);
	read(n);read(m);scanf("\n%s",maze+1);
	for(int i=1;i<=n;i++)col[i]=(maze[i]=='B');
	for(int i=1;i<=m;i++){
		int u,v;read(u);read(v);
		addEdge(u,v,i);addEdge(v,u,i);
	}
	for(int i=1;i<=n;i++)
		if(!belong[i]){
			belong[i]=++cnt,rt[cnt]=i;
			dfs1(i,0);dfs2(i,0);
		}
	for(int i=1;i<=n;i++)mf[belong[i]]=max(mf[belong[i]],f[i]);
	for(int i=1;i<=n;i++)sumf[belong[i]]+=1ll*f[i];
	for(int i=1;i<=cnt;i++)sumf[i]/=2LL;
	if(cnt==2)printf("%lld\n",sumf[1]+sumf[2]+1ll*mf[1]*sz[2]+1ll*mf[2]*sz[1]+1ll*sz[1]*sz[2]);
	else{
		LL res=0;
		res=max(res,(mf[2]+sz[2])*(n-sz[2])+(mf[3]+sz[3])*(n-sz[3])+getmx(rt[1],sz[2],sz[3]));
		res=max(res,(mf[1]+sz[1])*(n-sz[1])+(mf[2]+sz[2])*(n-sz[2])+getmx(rt[3],sz[1],sz[2]));
		res=max(res,(mf[1]+sz[1])*(n-sz[1])+(mf[3]+sz[3])*(n-sz[3])+getmx(rt[2],sz[1],sz[3]));
		printf("%lld\n",res+sumf[1]+sumf[2]+sumf[3]);
	}
	for(int i=1;i<=cnt;i++)dfs3(rt[i],0);int ans=0;
	for(int i=1;i<=cnt;i++){ 
		if(dp[rt[i]][col[rt[i]]]<0)ans=-1;
		if(ans<0)continue; 
		ans+=dp[rt[i]][col[rt[i]]];
	} 
	printf("%d\n",ans);if(ans<0)return 0;
	for(int i=1;i<=cnt;i++)dfs4(rt[i],0,col[rt[i]]);
	sort(sta+1,sta+stak+1);
	for(int i=1;i<=stak;i++)printf("%d ",sta[i]);puts("");
	return 0;
}

谢谢！！！

StaroForgin
关注
专栏目录
数据结构——树和森林
Labrador_Katie的博客
02-01 3349
树和森林树和森林树的存储结构双亲表示法 树和森林 树(Tree)：是n(n≥0)个结点的有限集；若n = 0，称为空树；其中R是称为根(Root)的结点 森林：是m(m≥0)棵互不相交的树的集合 树和森林是可以相互转换的，为森林增加一个根结点就变成了树，反之亦然 树的存储结构 双亲表示法 实现：定义结构数组存放树的结点，每个结点含两个域： 数据域：存放结点本身信息 双亲域：指示本结点的双亲结点在数组中的位置 特点：找双亲容易，找孩子难 类型描述： typedef struct PTNode{ T
6.4 树和二叉树-树和森林
michael_linux的博客，一个小小小学生。滴水穿石，步步为营，只为那刹那芳华。
08-01 1338
1. 树和森林的表示方法 树的三种存储结构 1.1 双亲表示法 #define MAX_TREE_SIZE 100 //结点结构: typedef struct PTNode{ Elem data; int parent; //双亲位置域 }PTNode; //树结构: typedef struct{ PTNode nodes[MAX_TREE_SIZE]; int root; //根结点的位置 int n; //结点个数 }PTree; 1.2 孩子链表表示法 //孩子结点结构 typede
树和森林的遍历
薛定谔的猫的博客
07-26 920
文章目录树和森林的遍历树的遍历树的先根遍历树的后根遍历层次遍历森林的遍历森林的先序遍历森林的中序遍历 树和森林的遍历 首先给出树和森林的遍历与二叉树遍历的对应关系： 树 森林 二叉树 先根遍历 先序遍历 先序遍历 后根遍历 中序遍历 中序遍历 树和森林的遍历均可采用对应二叉树的遍历算法来实现。 本文只简单出给树和森林的遍历方法思想及实例。 树的遍历  树的遍历操作就是按照某种方式访问树中的每个结点，且仅访问一次。树的遍历操作主要有先根遍历和后根遍历两种方式，此外还有层次遍历。 树
树和森林的概述
kcy的博客
11-18 6612
文章目录前言一、树的存储1.双亲表示法2.孩子表示法3.孩子兄弟表示法二、树，森林与二叉树之间的转化1.树转化成二叉树2.森林转换为二叉树3.二叉树转化成森林三、树和森林的遍历1.树的遍历(1)树的遍历    ①先根遍历：    ②后根遍历：(2)树的遍历算法2.森林的遍历 前言 在学了二叉树，这种特殊的树后，我们再回到一般的树，讨论他的存储和遍历 一、树的存储 实际在写代码的时候，人们用到很多种办法来存储树，以下我来介绍最
树和森林的区别
FengNingXueChen的博客
08-09 6381
1.树的存储结构    树有三种存储结构：双亲表示法，孩子表示法，兄弟表示
树和森林（快来瞧）
最新发布
qq_62687015的博客
01-20 1460
是由多颗互不相交的树所构成的树的集合，即森林包含多棵树，每一棵树都有自己的根结点。一棵树也可以看成森林。
数据机构之树和森林
梦的灰色边沿...
01-07 3262
树和森林一、树和森林的定义和性质1、树的定义　　树是n个结点的有限集合。 　　树可以为空树，即结点数为零。并且任意一颗非空树有以下特点: 　　1. 有且仅有一个根结点; 　　2. 当树的结点数大于1时，其余结点可分为多个结点的有限集合，这些集合的每一个又是一棵树，因此为根结点的子树。2、树中的相关术语　　假如有一棵树如下图所示: 　　祖先结点:从根节点到某个节点的路径上的所有节点都是该结点的
6.5 树和森林的遍历
嗑盐工作室
04-17 1867
6.5 树和森林的遍历 -树的遍历： ·先根遍历 ·后根遍历 A E G B C D F H I J 森林的遍历; 先序遍历和中序遍历（按后根遍历的方式遍历每一颗树） （1）中序遍历森林的第一课树的根节点的子树...
树和森林(Tree and Forest)
云原生Java Web领域技术博客
12-11 3291
树型结构是一种重要的非线性数据结构。树型结构在客观世界广泛存在，如组织关系可用树来表示。树在计算机领域也有广泛应用，如在编译程序时，可用树来表示源程序的语法结构(语法树)。又如在数据库系统中，使用树型结构存储索引等信息。森林则是零到多棵树的集合。对森林的研究，都是转化为对树的研究。
头歌 educoder 树和森林实验
01-13
在IT行业中，树和森林是数据结构中的基本概念，它们广泛应用于算法设计和问题解决。在"头歌 Educoder 树和森林实验"中，我们遇到了六个不同的挑战，每个挑战都涉及到了不同的知识点。让我们逐一深入探讨这些概念。 ...
合工大数据结构实验 树和森林
04-14
在本实验"合工大数据结构实验 - 树和森林"中，我们将深入探讨两种重要的数据结构：树和森林。 首先，我们要理解树的基本概念。树是一种非线性的数据结构，它由节点（也称为顶点）和边组成，每个节点可以有零个或多...
C++实现树和森林
08-25
1、 定义左儿子—右兄弟链接存储的树类和森林类。 2、 实验验证如下算法的正确性、各种功能及指标： 1）创建树和森林； 2）树和森林的先根遍历的递归和迭代算法； 3）树和森林的后根遍历的递归和迭代算法； 4）树和...
树和森林作业答案.zip
12-05
本压缩包“树和森林作业答案.zip”包含了一个名为“树和森林作业答案.ppt”的文件，我们可以推测这是一个关于树和森林数据结构的课后作业解答或讲解。下面将详细探讨树和森林的相关知识点。 首先，**树**是一种非...
[JOISC 2019 Day3]指定城市
StaroForgin的博客
04-27 2619
指定城市 题解 这题其实挺水的。 很明显，没被选的边的边权和最小也就是求出所有被选的边的边权和最大。 对于subtask2：也就是求只有一个点为指定点的情况。也就是求出所有点到他需要建的边。我们很容易求出对于一个固定点的所有儿子的子树上的点到他的和。这个一个dfs就可以解决。那么我们是否可以通过它父亲为root时的和求出它为root时的和吗？当然可以的。我们只需要再进行一次dfs，将它本身...
[CSP day1T3]树上的数
StaroForgin的博客
11-21 758
树上的数 题解 暴力(10pts) 一个的算法，枚举删边的顺序。 菊花图(25pts) 由于验题人的建议，出题人加上了这个部分。其实挺好想的，因为图中只有一个度为的点，其它点度数都为1。我们可以贪心枚举每个数字到达的点，用并查集将经过边定在一起即可。 链(25pts) **链。对于在链上的任意一条路径，它起始点往外的部分一定在路径之后被删，而目标点往外的部分一定在路径之前被删。我们...
[CTSC2017]网络
StaroForgin的博客
04-06 662
题外话，为什么我每次做题总能看到一个叫Claris的巨佬已经通过了此题。当然，ClariS唱地很好。
[2016-2017 ACM-ICPC CHINA-Final G]Pandaria
StaroForgin的博客
10-16 641
区间众数也太恶心了。
[AGC008F]Black Radius
StaroForgin的博客
02-05 606
OneInDark怎么天天爆切这种难难题呀！
树和森林的表示与遍历详解
本资源主要介绍了树和森林的表示方法以及遍历策略，包括双亲表示法、孩子链表表示法、孩子兄弟表示法，并强调了树的遍历及其应用。 在数据结构中，树是一种非线性数据结构，森林是若干棵树的集合。对于树的表示方法...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值