@Override public boolean contains(Object o) { T element = (T) o; Node<T> x = root; int cmp; while (x != null && (cmp = element.compareTo(x.key)) != 0) { if (cmp < 0) { x = x.left; } else { x = x.right; } } return x != null; }
时间: 2024-01-03 07:02:54 浏览: 68
intent传递ArrayList<T>
好的,以下是一个可能的测试类,可以用于测试上面的代码:
```
import static org.junit.Assert.*;
import org.junit.Before;
import org.junit.Test;
public class BinaryTreeContainsTest {
private Node<Integer> root;
@Before
public void setUp() throws Exception {
// 构造一棵二叉查找树
root = new Node<>(5);
root.left = new Node<>(3);
root.left.left = new Node<>(1);
root.left.right = new Node<>(4);
root.right = new Node<>(7);
root.right.left = new Node<>(6);
root.right.right = new Node<>(9);
}
@Test
public void testContains() {
assertTrue(contains(5));
assertTrue(contains(3));
assertTrue(contains(1));
assertTrue(contains(7));
assertTrue(contains(9));
assertFalse(contains(0));
assertFalse(contains(2));
assertFalse(contains(8));
}
private boolean contains(Integer element) {
Node<Integer> x = root;
int cmp;
while (x != null && (cmp = element.compareTo(x.key)) != 0) {
if (cmp < 0) {
x = x.left;
} else {
x = x.right;
}
}
return x != null;
}
}
```
这个测试类中,我们首先构造了一棵二叉查找树,然后使用 `contains()` 方法来判断树中是否包含某个元素,最后使用 JUnit 的断言方法 `assertTrue()` 和 `assertFalse()` 来判断判断结果是否正确。如果测试通过,那么说明上面的代码实现是正确的。
阅读全文