P5018 对称二叉树

对称二叉树(from luogu)
题目描述
一棵有点权的有根树如果满足以下条件，则被轩轩称为对称二叉树：

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

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

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

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

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

接下来 n 行，每行两个正整数 li,ri，分别表示节点 ii 的左右孩子的编号。如果不存在左 / 右孩子，则以 −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
说明/提示
【输入输出样例 1 说明】

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

【输入输出样例 2 说明】

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

数据规模与约定】
共 25 个测试点。
vi≤1000。
测试点 1∼3,n≤10，保证根结点的左子树的所有节点都没有右孩子，根结点的右 子树的所有节点都没有左孩子。
测试点4∼8,n≤10。
测试点9∼12,n≤10^5，保证输入是一棵“满二叉树” 。
测试点13∼16,n≤10 ^5，保证输入是一棵“完全二叉树”。
测试点17∼20,n≤10^5 ，保证输入的树的点权均为 1。
测试点21∼25,n≤10^6。

本题约定：

层次：节点的层次从根开始定义起，根为第一层，根的孩子为第二层。树中任一节 点的层次等于其父亲节点的层次加 1。

树的深度：树中节点的最大层次称为树的深度。

满二叉树：设二叉树的深度为 h，且二叉树有 2h−1 个节点，这就是满二叉树。

完全二叉树：设二叉树的深度为 h，除第 h 层外，其它各层的结点数都达到最大 个数，第 h 层所有的结点都连续集中在最左边，这就是完全二叉树。

题目分析
根据题意，我们不难理解，符合题意的对称有2种：结构上对称以及权值上对称
![在这里插入图片描述](https://img-blog.csdnimg.cn/20191102223956484.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2NfdWl6cnBfZHpqb3BrbA==,size_16,color_FFFFFF,t_70
如上图
如果一颗二叉树的对称的，那么我们需要保证：
L 与 R 对应
L.l 与 R.r 对应
L.r 与 R.l 对应
都是对称的（结构+权值，也就是说，最后一层都没有子节点）
通过观察，我们需要保证每一个子树的子树都是对称的

根据树的特点，我们不难想到DFS，
——我们可以用DFS找出那些点可以作为一颗对称子树（或者一整棵）的点的id同时计算出其子节点的个数
同时根据上述观点，我们不难得到一组可以判断不对称的依据：
①有不对称的子节点（对应的点之间）
②有不对等的权值
同时也可以得到正常退出的条件——都是叶子节点

代码

#include <bits/stdc++.h>
using namespace std;

struct node {
	int l;
	int r;
}num[1000009];

int n;
int v[1000009];
bool vis[1000009];
bool j;
int cnt,ans;

void search(int L,int R) {	
	if (L == -1 && R == -1) return ;   //正常的退出
	
	if ((L == -1 ^ R == -1) || (v[L] != v[R])) { j = false; return ; }   //当前选择的临时根节点的子树不对称
	
	cnt += 2;   //无论如何，如果对称，每一次都可以得到2个节点
	//对应的点——L.l对R.r；L.r对R.l —— 转化为下一次的L与R
	search(num[L].l,num[R].r);   
	search(num[L].r,num[R].l);
}

int main() {
	scanf("%d",&n);
	for (int i = 1; i <= n; i++) scanf("%d",&v[i]);
	for (int i = 1; i <= n; i++) scanf("%d %d",&num[i].l,&num[i].r);
	
	for (int i = 1; i <= n; i++) {
		j = true; cnt = 1;   //无论如何，肯定是有一个叶子节点当做对称
		search(num[i].l,num[i].r);
		if (j) ans = max(ans,cnt);
	}
	
	printf("%d",ans);
	
	return 0;
} 

crx CSP-J/S RP++