LeetCode110. 平衡二叉树

给定一个二叉树，判断它是否是高度平衡的二叉树。

本题中，一棵高度平衡二叉树定义为：

一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过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 。

题目大意：判断一棵二叉树是否是平衡的。

题目分析：通过dfs先序遍历二叉树的节点，同时记录节点的高度，一旦发现一个节点的左右子树高度差超过1，就可以判断该二叉树是非平衡的。

代码展示：

class Solution {
public:
    bool isBalanced(TreeNode* root) {
        if(root==NULL)
            return true;
        bool flag = true;
        dfs(root,flag);
        return flag;
    }
    
    int dfs(TreeNode* root,bool& flag){
        if(flag==false)
            return -1;
        if(root==NULL)
            return 0;
        int left = dfs(root->left,flag);
        int right = dfs(root->right,flag);
        if(abs(left-right)>1)
            flag = false;
        return max(left,right)+1;
    }
};