Leetcode: Balanced Binary Tree

平衡二叉树判断算法

最新推荐文章于 2025-09-10 15:24:29 发布

转载最新推荐文章于 2025-09-10 15:24:29 发布 · 48 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/EdwardLiu/p/3704786.html

文章标签：

#数据结构与算法

本文介绍了一种高效的判断二叉树是否为高度平衡的方法。通过递归算法结合巧妙的-1标记法，该方法仅需一层递归，运行时间为O(N)，空间复杂度为O(H)。

Given a binary tree, determine if it is height-balanced.

For this problem, a height-balanced binary tree is defined as a binary tree in which the depth of the two subtrees of every node never differ by more than 1.

很锻炼DP/recursive思路的一道题，个人感觉DP/recursive算是比较难写的题目了。这道题解法的巧妙之处在于巧用-1，并且使用临时存储，节省了很多开支。这道题同时也在Career Cup上面出现过

这道题我两次调试通过，第一次错是因为input{}, output false, expected true

我的算法只有一层递归(因为巧用-1的原因)，runs in O(N) time, andO(H) space

 1 public class Solution {
 2     public boolean isBalanced(TreeNode root) {
 3         if (root == null) return true;
 4         if (checkBalance(root) != -1) return true;
 5         else return false;
 6     }
 7     
 8     public int checkBalance(TreeNode root) {
 9         if (root == null) return 0;
10         int leftHeight = checkBalance(root.left);
11         int rightHeight = checkBalance(root.right);
12         if (leftHeight == -1 || rightHeight == -1) return -1;
13         else if (Math.abs(leftHeight - rightHeight) > 1) return -1;
14         return Math.max(leftHeight, rightHeight) + 1;
15     }
16 }