[Noip2018普及组]对称二叉树

最新推荐文章于 2023-08-01 19:08:16 发布

赵崇延

最新推荐文章于 2023-08-01 19:08:16 发布

阅读量616

点赞数 1

本文链接：https://blog.csdn.net/Zhao_Chongyan/article/details/86741289

版权

本文介绍了一种使用C++实现的检查二叉树是否对称的算法。通过对二叉树进行深度优先搜索，计算每个节点的子树大小，并检查左右子树是否相同来判断树的对称性。文章详细展示了算法的实现代码，包括定义节点结构、递归计算子树大小和检查子树对称性的过程。

摘要由CSDN通过智能技术生成

#include<cstdio>
#define max(a,b) ((a)>(b)?(a):(b))
const int MAXN = 1000000;
int siz[MAXN + 5], le[MAXN + 5], ri[MAXN + 5], v[MAXN + 5], ans;
bool check(int r1, int r2) {
    if( v[r1] != v[r2] ) return false;
    else if( r1 == 0 && r2 == 0 ) return true;
    else return check(le[r1], ri[r2]) && check(ri[r1], le[r2]);
}
int dfs1(int rt) {
    if( !rt ) return 0;
    else return siz[rt] = dfs1(le[rt]) + dfs1(ri[rt]) + 1;
}
void dfs2(int rt) {
    if( !rt ) return ;
    if( siz[le[rt]] == siz[ri[rt]] )
        if( check(le[rt], ri[rt]) ) ans = max(ans, siz[rt]);
    dfs2(le[rt]), dfs2(ri[rt]);
}
int main() {
    int n; v[0] = -1;
    scanf("%d", &n);
    for(int i=1;i<=n;i++)
        scanf("%d", &v[i]);
    for(int i=1;i<=n;i++) {
        scanf("%d%d", &le[i], &ri[i]);
        if( le[i] == -1 ) le[i] = 0;
        if( ri[i] == -1 ) ri[i] = 0;
    }
    dfs1(1), dfs2(1);
    printf("%d\n", ans);
}