110.平衡二叉树 | 分治思想

兀坐晴窗独饮茶

已于 2023-05-17 08:59:12 修改

阅读量590

点赞数

分类专栏：算法刷题文章标签： leetcode

于 2023-01-13 23:58:35 首次发布

本文链接：https://blog.csdn.net/weixin_40040107/article/details/128681136

版权

算法刷题专栏收录该内容

16 篇文章 0 订阅

订阅专栏

文章介绍了如何判断一棵二叉树是否是平衡二叉树，主要通过两种方法实现：分治策略和迭代的层序遍历。分治方法中，递归获取每个节点子树的高度，如果子树高度差超过1则返回-1表示不平衡；迭代方法则是利用层序遍历结合节点深度进行判断。这两种方法都在LeetCode的题目中有所体现。

摘要由CSDN通过智能技术生成

平衡二叉树

https://leetcode.cn/problems/balanced-binary-tree/submissions/

分治基本思路

一棵树是否是平衡二叉树, 是取决与它每个节点对应的子树是否是平衡二叉树, 只要有一个节点对应的子树不是, 这棵树就不是平衡二叉树

平衡二叉树 : 每个节点的左右两个子树的高度差的绝对值不超过 1

基本思路其实很简单 :

平衡二叉树需要判断每个节点的子树是否是平衡二叉树(大问题拆成小问题), 所以, 我们需要遍历二叉树
当前节点的高度, 其实就是当前节点到叶子节点的最大深度
所以我们获取高度以后, 在后序位置进行判断

分治代码实现

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {

    boolean isBalance = true;

    public boolean isBalanced(TreeNode root) {
        return getHeight(root) == -1 ? false : true;
    }
    
    // 如果是平衡二叉树就返回高度, 不是就返回 -1
    public int getHeight(TreeNode root) {

        if(root == null) {
            return 0;
        }

        int leftHeight = getHeight(root.left);
        // 一棵树是否是平衡二叉树, 是取决与它每个节点对应的子树是否是平衡二叉树, 
        // 只要有一个节点对应的子树不是, 这棵树就不是平衡二叉树
        
        if(leftHeight == -1) {
            return -1;
        }
        
        int rightHeight = getHeight(root.right);
        
        if(rightHeight == -1) {
            return -1;
        }

        if(Math.abs(leftHeight - rightHeight) > 1) {
            // 不是
            return -1;
        }

        return 1 + Math.max(leftHeight , rightHeight);
    }
   
}