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

题目

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

限制

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

思路

我们可以定义一个全局变量的答案，递归去做，一旦有某节点左右高度差大于1，我们直接将答案赋为false

代码

/**
 * 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 res = true;
    bool isBalanced(TreeNode* root) {
        dfs(root);
        return res;
    }
    int dfs(TreeNode* root) {
        if (!root) return 0;
        int l = dfs(root -> left),r = dfs(root -> right);
        if (abs(l - r) > 1) res = false;
        return max(l, r) + 1;
    }
};