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

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

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

 

示例 1:

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

    3
   / \
  9  20
    /  \
   15   7

返回 true 。

---

通过高度进行判断，和50-I差不多

 

class Solution {
public:
    bool flag = true;
    bool isBalanced(TreeNode* root) {
        Pan(root, 0);
        return flag;
    }

    int Pan(TreeNode* root, int num) {
        if (flag) {
            if (root == NULL)
                return num;
            int nl = Pan(root->left, num + 1);
            int nr = Pan(root->right, num + 1);
            if (abs(nl - nr) > 1)
                flag = false;
            return  nl > nr ? nl : nr;
        }
        return 0;
    }
};