洛谷 P5018 对称二叉树

本题重点是如何判断一棵二叉树是否是对称二叉树，它要满足以下条件：

1：结构相同，如果两棵树都为空树，则他们是对称二叉树，如果一棵非空，一棵空树，那么就不是对称二叉树。

2：权值相同，只需比较两个结点的权值。

由于是对称二叉树，因此我们递归判断左子树和右子树是否对称

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+2;
struct node{
	int left,right,val;
}tr[N];//采用结构体存储
bool same(int now1,int now2){
	if(now1==-1&&now2==-1) return true;//都为空树
	if(now1==-1||now2==-1) return false;//一棵空一棵非空
	if(tr[now1].val!=tr[now2].val) return false;//权值不同
	return same(tr[now1].left,tr[now2].right)&&same(tr[now1].right,tr[now2].left);
//递归判断左右子树相等，由于是对称二叉树，因此要左子树的左子树和右子树的右子树比较。
}
int count(int now){//计算一棵子树上的结点总数
	if(now==-1){
		return 0;
	}
	return 1+count(tr[now].left)+count(tr[now].right);
}
int main(){
	int n,ans=0;
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%d",&tr[i].val);
	}
	for(int i=1;i<=n;i++){
		scanf("%d %d",&tr[i].left,&tr[i].right);
	}
	for(int i=1;i<=n;i++){//枚举每棵子树，i为该子树的根节点
		if(same(i,i)){//该子树是对称二叉树
			ans=max(ans,count(i));
		}
	}
	printf("%d\n",ans);
	return 0;
}