剑指Offer面试题55 - II. 平衡二叉树

Posted 2020-11-27 galaxy-hao

题目

输入一棵二叉树的根节点，判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过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 <= 树的结点个数 <= 10000

本题同【LeetCode】110. 平衡二叉树

思路一：自顶向下

代码

时间复杂度：O(n^2)，每个点访问两次
空间复杂度：O(nlogn)

class Solution {
public:
    bool isBalanced(TreeNode* root) {
        if (!root) return true;
        int left = helper(root->left);
        int right = helper(root->right);
        if (abs(left - right) > 1) return false;
        return isBalanced(root->left) && isBalanced(root->right);
    }

    int helper(TreeNode *root) {
        if (!root) return 0;
        return max(helper(root->left), helper(root->right)) + 1;
    }
};