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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
import org.junit.jupiter.api.Test;
import static org.junit.jupiter.api.Assertions.*;
class TreeTraversalTest {
@Test
void test() {
TreeTraversal.TreeOrders tt = new TreeTraversal.TreeOrders();
tt.key = new int[]{4, 2, 5, 1, 3};
tt.left = new int[]{1, 3, -1, -1, -1};
tt.right = new int[]{2, 4, -1, -1, -1};
assertEquals(5, tt.inOrder().size());
assertEquals(1, tt.inOrder().get(0));
assertEquals(2, tt.inOrder().get(1));
assertEquals(3, tt.inOrder().get(2));
assertEquals(4, tt.inOrder().get(3));
assertEquals(5, tt.inOrder().get(4));
assertEquals(5, tt.preOrder().size());
assertEquals(4, tt.preOrder().get(0));
assertEquals(2, tt.preOrder().get(1));
assertEquals(1, tt.preOrder().get(2));
assertEquals(3, tt.preOrder().get(3));
assertEquals(5, tt.preOrder().get(4));
}
@Test
void test1() {
TreeTraversal.TreeOrders tt = new TreeTraversal.TreeOrders();
tt.key = new int[]{0, 10, 20, 30, 40, 50, 60, 70, 80, 90};
tt.left = new int[]{7, -1, -1, 8, 3, -1, 1, 5, -1, -1};
tt.right = new int[]{2, -1, 6, 9, -1, -1, -1, 4, -1, -1};
assertEquals(10, tt.inOrder().size());
assertEquals(50, tt.inOrder().get(0));
assertEquals(70, tt.inOrder().get(1));
assertEquals(80, tt.inOrder().get(2));
assertEquals(30, tt.inOrder().get(3));
assertEquals(90, tt.inOrder().get(4));
assertEquals(40, tt.inOrder().get(5));
assertEquals(0, tt.inOrder().get(6));
assertEquals(20, tt.inOrder().get(7));
assertEquals(10, tt.inOrder().get(8));
assertEquals(60, tt.inOrder().get(9));
assertEquals(10, tt.preOrder().size());
assertEquals(0, tt.preOrder().get(0));
assertEquals(70, tt.preOrder().get(1));
assertEquals(50, tt.preOrder().get(2));
assertEquals(40, tt.preOrder().get(3));
assertEquals(30, tt.preOrder().get(4));
assertEquals(80, tt.preOrder().get(5));
assertEquals(90, tt.preOrder().get(6));
assertEquals(20, tt.preOrder().get(7));
assertEquals(60, tt.preOrder().get(8));
assertEquals(10, tt.preOrder().get(9));
}
}
|