力扣题解-110. 平衡二叉树

题目: 110. 平衡二叉树

给定一个二叉树，判断它是否是高度平衡的二叉树。
本题中，一棵高度平衡二叉树定义为：
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过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 。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/balanced-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题解

树的高度即为树的最大深度, 因此此问题是求解二叉树的最大深度的变形. 求解二叉树最大深度的在这里力扣刷题-104. 二叉树的最大深度. 根据平衡二叉树的定义可知, 我们要想确定一个二叉树是否是平衡的, 需要确认每个节点的左右两个子树的高度差的绝对值不超过1. 在计算节点的左右两个子树的高度时, 用到了求解树的最大深度的过程.
因此, 同样的, 确定一颗二叉树是否为平衡二叉树也可以通过套用分治法的思想Divide and Conquer思想及实际应用来分析问题. 直接分治法三步走:

1. devide: 将原二叉树分解为根节点和左右两个子二叉树.
2. conquer: 首先判断根节点是否平衡, 然后再递归地判断左二叉树和右二叉树是否平衡.
3. combine: 将根节点, 左子树, 右子树的结果取并, 即得到原二叉树是否是平衡的.

代码

代码如下:

/**
 * 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 leftDepth = maxDepth(root->left);
        int rightDepth = maxDepth(root->right);
        int diff = leftDepth - rightDepth;
        if (diff > 1 or diff < -1) {
            return false;
        }
        return isBalanced(root->left) and isBalanced(root->right);
    }
    int maxDepth(TreeNode* root) {
        if (!root) {
            return 0;
        }
        int leftDepth = maxDepth(root->left);
        int rightDepth = maxDepth(root->right);
        return max(leftDepth, rightDepth) + 1;
    }
};