P5018 对称二叉树
题目传送门
这是一道经典的递归题,千万别被这题目给吓到了。
思路:
-
二叉树中的任意节点,都可能是一棵对称二叉树的根节点,那么就直接枚举1~n做根就行了。
-
从枚举的根开始,如果要生成对称二叉树,那左右子节点的结构及点值必须相同,接下来对于左右子节点的子节点也有相同的问题,那就是(如果要生成对称二叉树,那左右子节点的结构及点值必须相同)……。递归
#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
#include<string>
#include<algorithm>
#include<vector>
#define fre(x) freopen(#x".in","r",stdin),freopen(#x".out","w",stdout);
using namespace std;
const int MAX=2147483647;
const int N=1e6+5;
int n,ans=1,v[N],l[N],r[N];
bool f;
int dfs(int x,int y,int z)
{
if(x==-1&&y==-1) return 0; //无节点
if((x==-1||y==-1)||(v[x]!=v[y])) //结构或权值不对称
{
f=1; //标识不对称
return 0;
}
return dfs(l[x],r[y],2)+dfs(r[x],l[y],2)+z;
//继续往下以对称的形式递归:返回值为下面的节点数加上上面的节点数
}
int read()
{
char s;
int x=0,f=1;
s=getchar();
while(s<'0'||s>'9')
{
if(s=='-') f=-1;
s=getchar();
}
while(s>='0'&&s<='9')
{
x=(x<<3)+(x<<1)+s-'0';
s=getchar();
}
return x*f;
}
void input()
{
scanf("%d",&n);
for(int i=1;i<=n;i++) v[i]=read();
for(int i=1;i<=n;i++) l[i]=read(),r[i]=read();
for(int i=1;i<=n;i++)
{
int temp=dfs(l[i],r[i],3);
//从访问节点的左孩子和右孩子开始访问,故初始节点为三个
if(!f) ans=max(ans,temp);
f=0;
}
}
int main()
{
//fre(tree);
input();
printf("%d",ans);
return 0;
}