NOIP2018对称二叉树(树形结构的基础处理)

题目描述

传送门

一棵有点权的有根树如果满足以下条件,则被轩轩称为对称二叉树：

1.二叉树；
2.将这棵树所有节点的左右子树交换,新树和原树对应位置的结构相同且点权相等。
下图中节点内的数字为权值，节点外的$id$表示节点编号。

现在给出一棵二叉树，希望你找出它的一棵子树，该子树为对称二叉树，且节点数最多。请输出这棵子树的节点数。

注意：只有树根的树也是对称二叉树。本题中约定，以节点T为子树根的一棵“子树”指的是:节点T和它的全部后代节点构成的二叉树。

输入格式

第一行一个正整数n,表示给定的树的节点的数目,规定节点编号$1\sim n$,其中节点1是树根。
第二行n个正整数,用一个空格分隔,第i个正整数$v_i$ ,代表节点i的权值。
接下来n行,每行两个正整数$l_i,r_i$,分别表示节点i的左右孩子的编号.如果不存在左/右孩子,则以−1表示.两个数之间用一个空格隔开.

输出格式

输出文件共一行，包含一个整数，表示给定的树的最大对称二叉子树的节点数。

输入输出样例
输入 #1
2
1 3
2 -1
-1 -1
输出 #1
1
输入 #2
10
2 2 5 5 5 5 4 4 2 3
9 10
-1 -1
-1 -1
-1 -1
-1 -1
-1 2
3 4
5 6
-1 -1
7 8
输出 #2
3

思路

本题属于普及组2018最后一题(水题 ),(然而还没有T3的DP难…)
主要考察对树形结构的基本处理,我们来对此题进行分步解决.

1.预处理(建树+找根).
建树很简单我就不多说了(输完输入即完),然后我们将作为儿子的点打上标记,最后扫一遍,扫到没打标记的就是根了.(具体实现看代码)

inline void Init()
{
	n=in;
	for(int i=1;i<=n;i++)
	t[i].val=in;
	for(int i=1;i<=n;i++)
	{
		t[i].l=in,t[i].r=in;//输入节点i的左右儿子
		if(t[i].l>0)rd[t[i].l]++;//是节点就打标记
		if(t[i].r>0)rd[t[i].r]++;
	}
	for(int i=1;i<=n;i++)//找根
	{
		if(!rd[i]){
			root=i;
			break;
		}
	}
	return;
}

2.求每个子树的节点个数(设子树i的节点数为size[i])
用递归即可,即$size[i]=size[t[i].l]+size[t[i].r]+1$.
(不过儿子权值为-1时不能再往下递归)(代码很短qwq)

inline int Find(int k)
{
	size[k]=1;//自己是以自己为根所在子树的节点
	if(t[k].l!=-1)size[k]+=Find(t[k].l);//加上左子树节点数
	if(t[k].r!=-1)size[k]+=Find(t[k].r);//右子树节点数
	return size[k];
}

3.判断子树是符合要求.
这道题最有思维难度的便是写$check$函数,(其实也没什么难度 )
我们随便找一棵合法树进行分析,可以得到:
(1)当它的左右儿子均为空时，以它为根的子树显然合法.
(2)当它的左右儿子均不为空时,若左右儿子权值相等且左右儿子的儿子呈权值轴对称(说不清楚,还是看代码吧qwq ),那么这个子树同样合法.

inline bool check(int u,int v)
{
	if(u==-1&&v==-1)return true;
	if(u!=-1&&v!=-1&&t[u].val==t[v].val&&check(t[u].l,t[v].r)&&check(t[u].r,t[v].l))//判断合法
	return true;
	return false;
}

int main()
{
	for(int i=1;i<=n;i++)
	if(check(t[i].l,t[i].r)&&size[i]>ans)ans=size[i];//用ans计算最大值.
}

4.输出
不用我说吧…

printf("%d\n",ans);

完整代码如下:

#include<bits/stdc++.h>
#define N 1000006
#define in read()
using namespace std;

int n,rd[N],root,size[N],ans=1;
struct zb{
int l,r,val,size;}t[4*N];//树一般都得开4倍!

inline int in{
	int i=0,f=1;char ch;
	while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
	while(isdigit(ch)){i=(i<<3)+(i<<1)+(ch^48);ch=getchar();}
	return i*f;
}//很快的快读

inline void Init()
{
	n=in;
	for(int i=1;i<=n;i++)
	t[i].val=in;
	for(int i=1;i<=n;i++)
	{
		t[i].l=in,t[i].r=in;
		if(t[i].l>0)rd[t[i].l]++;
		if(t[i].r>0)rd[t[i].r]++;
	}
	for(int i=1;i<=n;i++)
	{
		if(!rd[i]){
			root=i;
			break;
		}
	}
	return;
}

inline int Find(int k)
{
	size[k]=1;
	if(t[k].l!=-1)size[k]+=Find(t[k].l);
	if(t[k].r!=-1)size[k]+=Find(t[k].r);
	return size[k];
}

inline bool check(int u,int v)
{
	if(u==-1&&v==-1)return true;
	if(u!=-1&&v!=-1&&t[u].val==t[v].val&&check(t[u].l,t[v].r)&&check(t[u].r,t[v].l))
	return true;
	return false;
}

int main()
{
	Init();
	Find(root);
	for(int i=1;i<=n;i++)
	if(check(t[i].l,t[i].r)&&size[i]>ans)ans=size[i];
	printf("%d\n",ans);
	return 0;
}

完结撒花~~~!