zl程序教程

您现在的位置是:首页 >  其他

当前栏目

Leetcode: Balanced Binary Tree

LeetCode Tree Binary Balanced
2023-09-11 14:14:08 时间
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 }