@@ -21,4 +21,37 @@ internal class SolutionTest {
2121 expected.add(listOf (5 , 6 ))
2222 assertThat(Solution ().levelOrder(root).toString(), equalTo(expected.toString()))
2323 }
24+
25+ @Test
26+ fun levelOrder2 () {
27+ val root = Node (1 )
28+ val node2 = Node (2 )
29+ val node3 = Node (3 )
30+ val node4 = Node (4 )
31+ val node5 = Node (5 )
32+ root.neighbors = listOf (node2, node3, node4, node5)
33+ val node6 = Node (6 )
34+ val node7 = Node (7 )
35+ node3.neighbors = listOf (node6, node7)
36+ val node11 = Node (11 )
37+ node7.neighbors = listOf (node11)
38+ val node14 = Node (14 )
39+ node11.neighbors = listOf (node14)
40+ val node8 = Node (8 )
41+ node4.neighbors = listOf (node8)
42+ val node12 = Node (12 )
43+ node8.neighbors = listOf (node12)
44+ val node9 = Node (9 )
45+ val node10 = Node (10 )
46+ node5.neighbors = listOf (node9, node10)
47+ val node13 = Node (13 )
48+ node9.neighbors = listOf (node13)
49+ val expected = mutableListOf<MutableList <Int >>()
50+ expected.add(mutableListOf<Int >(1 ))
51+ expected.add(mutableListOf<Int >(2 , 3 , 4 , 5 ))
52+ expected.add(mutableListOf<Int >(6 , 7 , 8 , 9 , 10 ))
53+ expected.add(mutableListOf<Int >(11 , 12 , 13 ))
54+ expected.add(mutableListOf<Int >(14 ))
55+ assertThat(Solution ().levelOrder(root).toString(), equalTo(expected.toString()))
56+ }
2457}
0 commit comments