题目描述
传送门
一棵有点权的有根树如果满足以下条件,则被轩轩称为对称二叉树:
1.二叉树;
2.将这棵树所有节点的左右子树交换,新树和原树对应位置的结构相同且点权相等。
下图中节点内的数字为权值,节点外的
i
d
{id}
id表示节点编号。
现在给出一棵二叉树,希望你找出它的一棵子树,该子树为对称二叉树,且节点数最多。请输出这棵子树的节点数。
注意:只有树根的树也是对称二叉树。本题中约定,以节点T为子树根的一棵“子树”指的是:节点T和它的全部后代节点构成的二叉树。
输入格式
第一行一个正整数n,表示给定的树的节点的数目,规定节点编号
1
∼
n
{1\sim n}
1∼n,其中节点1是树根。
第二行n个正整数,用一个空格分隔,第i个正整数
v
i
{v_i}
vi ,代表节点i的权值。
接下来n行,每行两个正整数
l
i
,
r
i
{l_i, r_i}
li,ri,分别表示节点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])
用递归即可,即
s
i
z
e
[
i
]
=
s
i
z
e
[
t
[
i
]
.
l
]
+
s
i
z
e
[
t
[
i
]
.
r
]
+
1
{size[i]=size[t[i].l]+size[t[i].r]+1}
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.判断子树是符合要求.
这道题最有思维难度的便是写
c
h
e
c
k
{check}
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;
}
完结撒花~~~!