429. N 叉树的层序遍历
N 叉树的层序遍历
题目描述:
给定一个 N 叉树,返回其节点值的层序遍历。(即从左到右,逐层遍历)。
树的序列化输入是用层序遍历,每组子节点都由 null 值分隔(参见示例)。
示例 1:
输入:root = [1,null,3,2,4,null,5,6]
输出:[[1],[3,2,4],[5,6]]
示例 2:
输入: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]
输出:[[1],[2,3,4,5],[6,7,8,9,10],[11,12,13],[14]]
提示:
树的高度不会超过 1000
树的节点总数在 [0, 10^4] 之间
思路:
时间复杂度:O(n),空间复杂度O(1)
用dfs最容易写了,不用栈之类的东西
代码:
/**
* Definition for a Node.
* type Node struct {
* Val int
* Children []*Node
* }
*/
func levelOrder(root *Node) [][]int {
var res [][]int
var dfs func(root *Node,dep int)
dfs = func(root *Node,dep int){
if root == nil{
return
}
if len(res) <= dep {
res = append(res, []int{})
res[dep] = append(res[dep], root.Val)
}else{
res[dep] = append(res[dep], root.Val)
}
for _,v := range root.Children{
dfs(v,dep+1)
}
}
dfs(root,0)
return res
}
代码效率:
执行用时:0 ms, 在所有 Go 提交中击败了100.00%的用户
内存消耗:4.1 MB, 在所有 Go 提交中击败了88.75%的用户
本博客所有文章除特别声明外,均采用 CC BY-SA 4.0 协议 ,转载请注明出处!