剑指Offer——JZ39.平衡二叉树【DFS】

最新推荐文章于 2024-01-12 23:20:25 发布

原创最新推荐文章于 2024-01-12 23:20:25 发布 · 327 阅读

0 ·

CC 4.0 BY-SA版权

剑指Offer 专栏收录该内容

36 篇文章

订阅专栏

本文介绍了一种用于判断二叉树是否平衡的算法。通过深度优先搜索(DFS)遍历树，比较每个节点的左右子树高度差，确保差值不超过1，以此判断整棵树的平衡性。代码实现采用C++，并提供了完整的AC-Code。

题目传送门

在这里插入图片描述

题解

所求即为：是否有节点的左右子树高度差 > 1
树上dfs跑一下即可

AC-Code

class Solution {
public:
    bool IsBalanced_Solution(TreeNode* pRoot) {
        if(!pRoot)    return true;
        return dfs(pRoot) != -1;
    }
    int dfs(TreeNode* now) {
        if(!now)    return 0;
        int lnum = 0, rnum = 0;	// 别忘了初始化
        if(now->left)
            lnum = dfs(now->left);
        if(now->right)
            rnum = dfs(now->right);
        if(abs(lnum - rnum) > 1 || lnum == -1 || rnum == -1) {
            return -1;
        }
        return max(lnum, rnum) + 1;
    }
};