平衡二叉树（Python and C++算法）

题目：

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

限制：

1 <= 树的结点个数 <= 10000

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/ping-heng-er-cha-shu-lcof

思路：

　　判断的过程中需要计算树的深度，以某节点为父节点的树深度=max(左子树的深度，右子树的深度)+1。

　　采用后序遍历的方法，从底至顶判断，先遍历左右子节点得出子节点的深度，判断当前子树是否符合要求：

　　　　如果符合要求则返回当前子树的深度；

　　　　如果不符合要求则返回-1，若发现某子树深度为-1，直接返回相当于进行剪枝。

Python解法：

class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

class Solution:
    def isBalanced(self, root: TreeNode) -> bool:
        def recur(root):
            if root is None:  # 递归终止条件
                return 0
            leftDepth = recur(root.left)  # 递归左子树
            if leftDepth == -1:
                return -1
            rightDepth = recur(root.right)  # 递归右子树
            if rightDepth == -1:
                return -1
            if abs(leftDepth-rightDepth) <= 1:
                return max(leftDepth, rightDepth) + 1
            else:
                return -1
        return recur(root) != -1

C++解法：

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
 
class Solution {
public:
    bool isBalanced(TreeNode* root) {
        return recur(root) != -1;
    }
    int recur(TreeNode *root) {
        if(root == NULL)
            return 0;
        int leftDepth = recur(root -> left);
        if(leftDepth == -1)
            return -1;
        int rightDepth = recur(root -> right);
        if(rightDepth == -1)
            return -1;
        if(abs(leftDepth - rightDepth) <= 1)
            return max(leftDepth, rightDepth) + 1;
        else
            return -1;
    }
};