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 协议 ,转载请注明出处!