leetcode_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 。

解析:本题要求判断给定的树是否高度平衡。
对于每一个节点我们需要做以下4步:
1、获取左节点的深度
2、获取右节点的深度
3、判断左右节点相差是否超过1
4、不超过1返回该节点的深度

从逻辑去想我们或许需要两种递归函数,一种用来获取深度,另一种用来判断高度是否平衡,但是这样递归次数太多太慢
我们可以利用负数,对于一个节点的深度,它最小为0,即空节点,因此我们可以定义如果节点的深度为负数则代表该节点高度不平衡,这样就可以获取深度的同时判断是否高度平衡

class Solution {
public:
    bool isBalanced(TreeNode* root) {
        return (nodeDepth(root)!=-1);
    }
    //判断该节点以及其左右子树各节点是否高度平衡,不平衡返回-1
    //平衡返回该节点深度
    int nodeDepth(TreeNode* root){
        if(!root) return 0; //空节点高度为0
        if(!root->left&&!root->right) return 1; //叶子节点高度为1
        //左右子树不符合要求层层返回
        int l = nodeDepth(root->left);
        int r = nodeDepth(root->right);
        if(l<0) return -1; 
        if(r<0) return -1;
        if(abs(l-r)>1) return -1;
        //均符合要求的情况下返回该节点的深度
        return max(l,r)+1;
    }
};

发布了146 篇原创文章 · 获赞 0 · 访问量 748
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 深蓝海洋 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览