Nary-Tree input serialization is represented in their level order traversal. Each group of children is separated by the null value (See examples)
Input: root = [1,null,3,2,4,null,5,6]
Output:
[1,3,5,6,2,4]
Input: root = [1,null,2,3,4,5,null,null,6,7,null,8,null,9,10,null,null,11,null,12,null,13,null,null,14]
Output:
[1,2,3,6,7,11,14,4,8,12,5,9,13,10]
/**
* Definition for a Node.
* type Node struct {
* Val int
* Children []*Node
* }
*/
func preorder(root *Node) []int {
res := make([]int, 0)
dfs(root, &res)
return res
}
func dfs(root *Node, arr *[]int) {
if root == nil {
return
}
*arr = append(*arr, root.Val)
for _, n := range root.Children {
dfs(n, arr)
}
}
/**
* Definition for a Node.
* type Node struct {
* Val int
* Children []*Node
* }
*/
func preorder(root *Node) []int {
stack := make([]*Node, 0)
res := make([]int, 0)
if root == nil {
return res
} else {
stack = append(stack, root)
}
for len(stack) != 0 {
node := stack[len(stack) - 1]
stack = stack[:len(stack) - 1]
res = append(res, node.Val)
for i:=len(node.Children) - 1; i >= 0; i-- {
stack = append(stack, node.Children[i])
}
}
return res
}