C/C++数据结构——[NOIP2018]对称二叉树（树状结构）

题目描述

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

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

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

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

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

树的深度:树中节点的最大层次称为树的深度。 满二叉树:设二叉树的深度为 h，且二叉树有 2h − 1 个节点，这就是满二叉树。

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

输入描述:

第一行一个正整数 𝑛，表示给定的树的节点的数目，规定节点编号 1~n，其中节点1 是树根。 第二行 𝑛 个正整数，用一个空格分隔，第 𝑖 个正整数 𝑣𝑖 代表节点 𝑖 的权值。 接下来 𝑛 行，每行两个正整数 𝑙 , 𝑟 ，分别表示节点 𝑖 的左右孩子的编号。如果不存在左 / 右孩子，则以 −1 表示。两个数之间用一个空格隔开。

输出描述:

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

示例1

输入

2
1 3
2 -1
-1 -1

输出

1

说明

最大的对称二叉子树为以节点 2 为树根的子树，节点数为 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

输出

3

说明

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

备注:

共 25 个测试点。 𝑣𝑖 ≤ 1000。
测试点 1~3，𝑛 ≤ 10，保证根结点的左子树的所有节点都没有右孩子，根结点的右子树的所有节点都没有左孩子。
测试点 4~8，𝑛 ≤ 10。
测试点 9~12，𝑛 ≤ 105，保证输入是一棵“满二叉树”。
测试点13~16，𝑛 ≤ 105，保证输入是一棵“完全二叉树”。
测试点 17~20，𝑛 ≤ 105，保证输入的树的点权均为 1。
测试点 21~25，𝑛 ≤ 106。
链接：https://ac.nowcoder.com/acm/problem/21472
来源：牛客网

解题代码

#include<stdio.h>
typedef struct Tree{
    int num;
    int l,r;
    int size;
}T;
T tree[1000005];
int maxLen=1;
int judgeFullTree(int l,int r)
{
	if(l==-1&&r==-1) return 1;
	if(l==-1) return 0;
	if(r==-1) return 0;
	if(tree[l].num!=tree[r].num)return 0;
	if(judgeFullTree(tree[l].l,tree[r].r)&&judgeFullTree(tree[l].r,tree[r].l)){
		return 1;
	}
    return 0;
}

void dfsTree(int i)
{
    tree[i].size=1;
    if(tree[i].l!=-1){
        dfsTree(tree[i].l);
        tree[i].size+=tree[tree[i].l].size;
    }
    if(tree[i].r!=-1){
        dfsTree(tree[i].r);
        tree[i].size+=tree[tree[i].r].size;
    }
}

int main()
{
    
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&tree[i].num);
    }
    for(int i=1;i<=n;i++){
        scanf("%d %d",&tree[i].l,&tree[i].r);
    }
	dfsTree(1);
	for(int i=1;i<=n;i++){
		if(judgeFullTree(tree[i].l,tree[i].r)){
			maxLen=tree[i].size>maxLen?tree[i].size:maxLen;
		}
	}
    printf("%d",maxLen);
}

解题思路

这道题，要注意什么时候才有对称的情况，无非就是满二叉树的时候。而满二叉树的条件就是二叉树的深度为 h，且二叉树有 2h − 1 个节点，这就是满二叉树，也就是一个根节点的左右结点都存在或者这是都不存在的时候我们才需要继续遍历。
树状结构去做会比较方便，节省了链树的构造树这个部分。刚开始是想到对这个树进行中序遍历，将结点值全部存储，然后去找其中的最大回文子串，但是这样找又好麻烦，增加了时间和空间。
其实我们只需要找每一个结点的时候，都对结点进行检查，查看子树是否符合满二叉树的情况且左右结点相等，那么继续递归检查，将子树长度记录然后存储，最终比较子树结点大小就行了。