[剑指Offer]:平衡二叉树

69 阅读1分钟

文章目录


题目描述

输入一棵二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1,那么它就是一棵平衡二叉树。

示例 1:

给定二叉树 [3,9,20,null,null,15,7]

    3
   / \
  9  20
    /  \
   15   7

返回 true 。

示例 2:

给定二叉树 [1,2,2,3,3,null,null,4,4]

       1
      / \
     2   2
    / \
   3   3
  / \
 4   4

返回 false 。

题解思路

方法一:先序遍历

先序遍历每一个节点,并比较左右子树高度,如果有>1则返回false

代码实现:

class Solution {
public:
    bool isBalanced(TreeNode* root) {
        if(!root) return true;

        return (abs(maxDepth(root->left) - maxDepth(root->right)) <= 1) && isBalanced(root->left) && isBalanced(root->right);
    }
private:
    int maxDepth(TreeNode* root){
        if(!root) return 0;
        return max(maxDepth(root->left), maxDepth(root->right)) + 1;
    }
};

方法二:后序遍历

  • 使用后序遍历的方式遍历二叉树的每个节点,那么在遍历到一个节点之前我们就已经
  • 遍历了它的左右子树。只要在每个节点的时候记录它的深度(某一节点的深度等于它到叶节点的路径的长度)
  • 就可以判断每个节点是不是平衡的

代码实现:

class Solution {
public:
    bool isBalanced(TreeNode* root) {
        if(!root) return true;
        int depth = 0;
        return isBalanced(root, depth);
    }
private:
    bool isBalanced(TreeNode* root, int& depth){
        if(root == nullptr){
            depth = 0;
            return true;
        }
        int l = 0, r = 0;
        if(isBalanced(root->left, l) && isBalanced(root->right, r)){
            int dif = abs(l - r);
            if(dif <= 1){
                depth = (l > r ? l : r) + 1;
                return true;
            }
        }
        return false;
    }
};

如有帮助到您,可以多多点赞、评论鼓励哟~~~