Skip to content

Commit 5a94621

Browse files
committed
Added test 429
1 parent e0402fc commit 5a94621

File tree

1 file changed

+33
-0
lines changed
  • src/test/kotlin/g0401_0500/s0429_n_ary_tree_level_order_traversal

1 file changed

+33
-0
lines changed

src/test/kotlin/g0401_0500/s0429_n_ary_tree_level_order_traversal/SolutionTest.kt

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)