码迷,mamicode.com
首页 > 其他好文 > 详细

Leetcode solution 110: Balanced Binary Tree

时间:2019-08-06 14:18:40      阅读:101      评论:0      收藏:0      [点我收藏+]

标签:cal   direct   ted   height   public   statement   with   class   pac   

 Problem Statement 

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.

Example 1:

Given the following tree [3,9,20,null,null,15,7]:

    3
   /   9  20
    /     15   7

Return true.

 

Example 2:

Given the following tree [1,2,2,3,3,null,null,4,4]:

       1
      /      2   2
    /    3   3
  /  4   4

Return false.

Problem link

 

Video Tutorial

You can find the detailed video tutorial here

 

Thought Process

As described in the problem, it is intuitive to solve this problem recursively, especially given this is a tree related problem. What we can do is get the height of the left sub tree, compared with the right sub tree, then do the logics to see if it’s balanced or not. If at certain level either the left or right sub tree is not balanced, then entire Tree is not balanced. Classic usage for post order traversal.

 

Solutions

 

 1 public boolean isBalanced(TreeNode root) {
 2         return maxHeight(root) != -1;
 3     }
 4     
 5     // if -1, means it is not a balanced tree, since it will also return the normal height(int), so boolean is not an option. 
 6     // Kinda of a hack for the return type.
 7     // @return -1 means it‘s already not a balanced tree, else t;he tree height
 8     public int maxHeight(TreeNode root) {
 9         if (root == null) {
10             return 0;
11         }
12         
13         int l = maxHeight(root.left);
14         int r = maxHeight(root.right);
15 
16         // a classic usage of post order traversalß
17         if (l == -1 || r == -1 || Math.abs(l - r) > 1) {
18             return -1;
19         }
20         return Math.max(l, r) + 1;
21     }

 

Post order traversal using recursion

Calculate the height of the left subtree, calculate the height of the right subtree, then compare. If it‘s already not balanced, return -1 and directly return

 

Time Complexity: O(N), N is the total tree nodes since it is visited once and only once

Space Complexity: O(1), No extra space is needed

 

References

Leetcode solution 110: Balanced Binary Tree

标签:cal   direct   ted   height   public   statement   with   class   pac   

原文地址:https://www.cnblogs.com/baozitraining/p/11308254.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!