fixed testInterval

This commit is contained in:
navid.sassan 2020-10-25 22:35:52 +01:00
parent 473c333cfd
commit 15dce99c64
2 changed files with 15 additions and 8 deletions

View File

@ -10,4 +10,5 @@ public interface Traversal<T> {
public void postorder(Visitor<T> vistor); public void postorder(Visitor<T> vistor);
/* traverse elements of tree in levelorder */ /* traverse elements of tree in levelorder */
public void levelorder(Visitor<T> vistor); public void levelorder(Visitor<T> vistor);
public void interval(Comparable<T> min, Comparable<T> max, Visitor<T> v);
} }

View File

@ -46,7 +46,8 @@ public class TreeTest {
@Test @Test
public void testInterval() { public void testInterval() {
Comparable<Integer> myComparator = new MyComparator(); Comparable<Integer> minComparable = new MyComparable(1);
Comparable<Integer> maxComparable = new MyComparable(9);
Visitor<Integer> v = new MyVisitor<Integer>(); Visitor<Integer> v = new MyVisitor<Integer>();
Tree<Integer> mytree = new SortedBinaryTree<Integer>(); Tree<Integer> mytree = new SortedBinaryTree<Integer>();
mytree.add(10); mytree.add(10);
@ -58,16 +59,21 @@ public class TreeTest {
mytree.add(8); mytree.add(8);
mytree.add(4); mytree.add(4);
tree.traversal().interval(myComparator, myComparator, v); mytree.traversal().interval(minComparable, maxComparable, v);
System.out.println(v.toString()); assertEquals("interval", "7648", v.toString());
// assertEquals("levelorder", "BACD", v.toString());
} }
} }
class MyComparator implements Comparable<Integer> { class MyComparable implements Comparable<Integer> {
public int compare(TreeNode o1, TreeNode o2) { Integer value;
return o1.element.compareTo(o1.element);
public MyComparable(Integer value) {
this.value = value;
}
@Override
public int compareTo(Integer other) {
return this.value.compareTo(other);
} }
} }