树形结构+思维 对称二叉树（洛谷 P5018）

对称二叉树

题目描述

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

二叉树；
将这棵树所有节点的左右子树交换，新树和原树对应位置的结构相同且点权相等。

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

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

输入格式
第一行一个正整数 n，表示给定的树的节点的数目，规定节点编号 1∼n，其中节点 1 是树根。

第二行 n 个正整数，用一个空格分隔，第 i 个正整数 vi 代表节点 i 的权值。

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

输出格式

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

---

找出所有子树，判断其是否对称（一颗树是否对称取决于它的所有子树都对称），找出最大的对称子树；

递归遍历树是显然的，刚开始想的是每颗树的左右子树权值能组成的字符串都算出来，然后判断左右子树的字符串是否相同，但是又TLE又WA；

正解是对于每个子树的根结点，其左右子树的结点假设为 l，r，那么应该满足
w[ L[l] ]=w[ R[r] ]&&w[ R[l] ]=w[ L[r] ] ;

#include<bits/stdc++.h>
#define LL long long
#define pa pair<int,int>
#define ls k<<1
#define rs k<<1|1
#define inf 0x3f3f3f3f
using namespace std;
const int N=1000010;
const int M=1000100;
const int mod=1e5+7;
int n,w[N],L[N],R[N],size[N],ans;
int dfs(int fa){
	size[fa]=1;
	if(L[fa]!=-1) size[fa]+=dfs(L[fa]);
	if(R[fa]!=-1) size[fa]+=dfs(R[fa]);
	return size[fa];
}
bool judge(int l,int r){
	if(l==-1&&r==-1) return true;
	if(l!=-1&&r!=-1&&w[l]==w[r]&&judge(L[l],R[r])&&judge(R[l],L[r])) return true;
	return false;
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++) scanf("%d",&w[i]);
	for(int i=1;i<=n;i++) scanf("%d%d",&L[i],&R[i]);
	dfs(1);
	for(int i=1;i<=n;i++){
		if(judge(L[i],R[i])){
			ans=max(size[i],ans);
		}
	}
	cout<<ans<<endl;
    return 0;
}