平衡二叉树

文章内容是自己刷leetcode题目的一些总结。
文章内容参考公众号： 代码随想录。
喜欢的话，希望大家可以点点赞 ^ - ^

一.平衡二叉树的介绍

一棵高度平衡二叉树定义为：一个二叉树每个节点的左右两个子树的高度差的绝对值不超过1。

二.判断平衡二叉树

由平衡二叉树的定义我们知道是要判断树的高度，二叉树的高度我们前面介绍过，就是该节点到叶子节点的的最远路径。

因为我们要比较高度，所以使用后序遍历，先递归左右子树，然后比较左右子树的高度差是否大于1。

单次递归逻辑是：如果左右子树高度差大于1，那么我们返回-1，证明以当前节点为根节点的树是非平衡的；如果高度差小于等于1，我们返回当前节点的高度。

三.判断平衡二叉树的递归代码

根据上述分析，我们由递归三部曲可以得到：

• 递归参数与返回值：只需要根节点作为参数，返回值为int类型，如果返回值为-1，说明树不是平衡的。
• 终止条件：遇到空节点返回0。
• 单次递归逻辑：使用后序遍历，递归左右子树，如果发现有一个子树不平衡，就返回-1；中间节点逻辑为左右子树高度差的绝对值大于1，树不平衡返回-1，否则平衡，返回当前树的高度。

class Solution {
public:
    int getHeight(TreeNode* node) {
        //后序遍历
        //终止条件
        if(node == nullptr) return 0;
        //单次逻辑
        int lHeight = getHeight(node->left);
        if(lHeight == -1) return -1;
        int rHeight = getHeight(node->right);
        if(rHeight == -1) return -1;
        return abs(lHeight - rHeight) > 1 ? -1 : 1 + max(lHeight, rHeight);
    }
    bool isBalanced(TreeNode* root) {
        //递归写法：
        //因为要考虑高度，所以用后序遍历
        return getHeight(root) == -1 ? false : true;
    }
};

四.总结

判断平衡二叉树其实是计算机专业课数据结构中的一个知识，需要判断节点左右子树的高度差来判断是否平衡，其中用到了二叉树高度的知识，高度是当前节点到子节点的最远距离，这与我们前面讲过的知识又串了起来。

希望可以帮助到大家。