diff --git a/06/src/main/java/ch/zhaw/ads/AVLSearchTree.java b/06/src/main/java/ch/zhaw/ads/AVLSearchTree.java index 57e636d..89022b1 100644 --- a/06/src/main/java/ch/zhaw/ads/AVLSearchTree.java +++ b/06/src/main/java/ch/zhaw/ads/AVLSearchTree.java @@ -1,5 +1,7 @@ package ch.zhaw.ads; +import java.lang.Math; + public class AVLSearchTree> extends SortedBinaryTree { // /** @@ -13,6 +15,14 @@ public class AVLSearchTree> extends SortedBinaryTree return calcHeight(root); } + public boolean balanced(TreeNode node) { + if (node == null) { + return true; + } else { + return Math.abs(calcHeight(node.left) - calcHeight(node.right)) < 2; + } + } + /** * Insert into the tree; duplicates are ignored. * @param element the item to insert.