-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathBalancedBinaryTree.java
More file actions
37 lines (33 loc) · 1.03 KB
/
Copy pathBalancedBinaryTree.java
File metadata and controls
37 lines (33 loc) · 1.03 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
public class BalancedBinaryTree {
public boolean isBalanced(TreeNode root) {
if(root == null) return true;
if(Math.abs(height(root.left) - height(root.right)) > 1) return false;
if(!isBalanced(root.left) || !isBalanced(root.right)) return false;
return true;
}
private int height(TreeNode root){
if(root == null) return 0;
return Math.max(height(root.left), height(root.right)) + 1;
}
public static void main(String[] args) {
BalancedBinaryTree tree = new BalancedBinaryTree();
TreeNode root = new TreeNode(5);
TreeNode node1 = new TreeNode(4);
TreeNode node2 = new TreeNode(8);
TreeNode node3 = new TreeNode(11);
TreeNode node4 = new TreeNode(13);
TreeNode node5 = new TreeNode(4);
TreeNode node6 = new TreeNode(7);
TreeNode node7 = new TreeNode(2);
TreeNode node8 = new TreeNode(1);
root.left = node1;
root.right = node2;
node1.left = node3;
node2.left = node4;
node2.right = node5;
node3.left = node6;
node3.right = node7;
node5.right = node8;
System.out.println(tree.isBalanced(root));
}
}