数组解决对称二叉树

#数组解决对称二叉树
##洛谷P5018
###题目
一棵有点权的有根树如果满足以下条件，则被轩轩称为对称二叉树：

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

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

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

Input
第一行一个正整数 nn，表示给定的树的节点的数目，规定节点编号 1 \sim n1∼n，其中节点 11 是树根。

第二行 nn 个正整数，用一个空格分隔，第 ii 个正整数 v_iv
i
​
代表节点 ii 的权值。

接下来 nn 行，每行两个正整数 l_i, r_il
i
​
,r
i
​
，分别表示节点 ii 的左右孩子的编号。如果不存在左 / 右孩子，则以 -1−1 表示。两个数之间用一个空格隔开。

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

Sample 1
Inputcopy Outputcopy
2
1 3
2 -1
-1 -1
1
Sample 2
Inputcopy Outputcopy
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
3
Hint
【输入输出样例 1 说明】

最大的对称二叉子树为以节点 22 为树根的子树，节点数为 11。

【输入输出样例 2 说明】

最大的对称二叉子树为以节点 77 为树根的子树，节点数为 33。

【数据规模与约定】
共 2525 个测试点。
v_i ≤ 1000v
i
​
≤1000。
测试点 1 \sim 3, n ≤ 101∼3,n≤10，保证根结点的左子树的所有节点都没有右孩子，根结点的右 子树的所有节点都没有左孩子。
测试点 4 \sim 8, n ≤ 104∼8,n≤10。
测试点 9 \sim 12, n ≤ 10^59∼12,n≤10
5
，保证输入是一棵“满二叉树” 。
测试点 13 \sim 16, n ≤ 10^513∼16,n≤10
5
，保证输入是一棵“完全二叉树”。
测试点 17 \sim 20, n ≤ 10^517∼20,n≤10
5
，保证输入的树的点权均为 11。
测试点 21 \sim 25, n ≤ 10^621∼25,n≤10
6
###代码

#include<iostream>
#include<cstring>
#include<algorithm>
#include<string>
using namespace std;
const int N=1000005;
int v[N],l[N],r[N];
int pan=1;
int ans=0;
void pd(int x,int y)
{
	if(x==-1&&y==-1){
		return;
	}
	if(x==-1||y==-1||v[x]!=v[y]){
		pan=0;
		return;
	}
	pd(l[x],r[y]);
	pd(r[x],l[y]);
}
int jishu(int x)
{
	int k=1;
	if(l[x]!=-1){
		k+=jishu(l[x]);
	}
	if(r[x]!=-1){
		k+=jishu(r[x]);
	}
	return k;
}
int main()
{
	int n;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>v[i];	
	}	
	for(int i=1;i<=n;i++)
	{
		cin>>l[i]>>r[i];
	}
	for(int i=1;i<=n;i++)
	{
		pan=1;
		pd(l[i],r[i]);
		if(pan==1){
			ans=max(ans,jishu(i));
		}
	}
	cout<<ans<<endl;
	return 0;
}