剑指-面试题55-2 平衡二叉树

平衡二叉树

题目
输入一棵二叉树的根节点，判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1，那么它就是一棵平衡二叉树。
$1<=树的结点个数<=10000$

思路
        对二叉树做后序遍历时，遍历到一个节点之前已经遍历了其左右子树。
        所以只要在遍历每个节点时记录其深度，从底至顶返回子树深度，若判定某子树不是平衡树则 “剪枝” ，直接向上返回。

• 当节点root 左 / 右子树的深度差≤1 ：则返回当前节点的深度，
• 当节点root 左 / 右子树的深度差 > 2 ：则返回 -1，代表此子树不是平衡树 。

时间复杂度$O(N)$ ： N 为树的节点数；最差情况下，需要递归遍历树的所有节点。
空间复杂度 $O(N)$： 最差情况下（树退化为链表时），系统递归需要使用 O(N) 的栈空间。

---

C++

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    bool isBalanced(TreeNode* root) {
        if(root==nullptr)
            return true;
        int depth = 0;
        return judge(root, depth);
    }
private:
    bool judge(TreeNode* root, int &depth)
    {
        if(root==nullptr)
        {
            depth = 0;
            return true;
        }
        int left_depth = 0;
        int right_depth = 0;
        if(judge(root->left, left_depth) && judge(root->right, right_depth))            //先判断左右，都满足了再判断根
        {
            int diff = left_depth-right_depth;
            if(abs(diff)<=1)
            {
                depth = max(left_depth, right_depth)+1;
                return true;
            }
        }
        return false;
    }
};

python

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def isBalanced(self, root: TreeNode) -> bool:
        def judge(root: TreeNode):
            if not root:
                return 0
            left = judge(root.left)
            if left == -1:
                return -1
            right = judge(root.right)              
            if right == -1:
                return -1
            if abs(left-right)<=1:
                return max(left, right)+1
            else:
                return -1
        return judge(root) != -1