0
点赞
收藏
分享

微信扫一扫

剑指Offer:[第18天 搜索与回溯算法(中等)]--->Ⅱ. 平衡二叉树

文章目录


一、题目描述

输入一棵二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过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 

限制:


二、使用步骤

注:思路分析中的一些内容和图片参考自力扣各位前辈的题解,感谢他们的无私奉献

思路:


三、整体代码

整体代码如下

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */

int height(struct TreeNode* root) {
    //如果为空结点,代表穿过叶节点,深度为0
    if (root == NULL) {
        return 0;
    }
    //递归处理左右子树
    int leftHeight = height(root->left);  
    int rightHeight = height(root->right);
    //如果左子树不平衡  或者右子树不平衡  或者左右子树深度差大于1
    //则代表当前结点不平衡,直接剪枝,返回-1
    if (leftHeight == -1 || rightHeight == -1 || fabs(leftHeight - rightHeight) > 1) {
        return -1;
    } else {
        //否则返回当前这棵树的深度
        return fmax(leftHeight, rightHeight) + 1;
    }
}

bool isBalanced(struct TreeNode* root) {
    return height(root) >= 0;
}

运行,测试通过
在这里插入图片描述

举报

相关推荐

0 条评论