前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Tree - 102. Binary Tree Level Order Traversal

Tree - 102. Binary Tree Level Order Traversal

作者头像
ppxai
发布2020-09-23 17:24:34
3210
发布2020-09-23 17:24:34
举报
文章被收录于专栏:皮皮星球

102. Binary Tree Level Order Traversal

Given a binary tree, return the level order traversal of its nodes' values. (ie, from left to right, level by level).

For example: Given binary tree [3,9,20,null,null,15,7],

代码语言:javascript
复制
    3
   / \
  9  20
    /  \
   15   7

return its level order traversal as:

代码语言:javascript
复制
[
  [3],
  [9,20],
  [15,7]
]

思路:

层序遍历二叉树,这一题和637题很像,637是求每一层的平均值,这一题只让打印出来,可以用bfs做,也可以使用dfs做,这里用bfs做,有递归写法和非递归写法,非递归写法使用了队列在存储下一层要遍历的节点。

代码:

go:

代码语言:javascript
复制
/**

 * Definition for a binary tree node.

 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
/*// bfs recursive
func levelOrder(root *TreeNode) [][]int {
    var res [][]int
    if root == nil { return res }
    
    traversalNode(root, &res, 0)
    
    return res
}

func traversalNode(node * TreeNode, res *[][]int, h int) {
    if node == nil { return }
    if len(*res) < h + 1 {
        *res = append(*res, []int{})
    }
    
    (*res)[h] = append((*res)[h], node.Val)
    traversalNode(node.Left, res, h+1)
    traversalNode(node.Right, res, h+1)
}*/


// bfs iterative
func levelOrder(root *TreeNode) [][]int {
    var res [][]int
    if root == nil {
        return res
    }
    queue := []*TreeNode{root}
    
    for len(queue) != 0 {
        var temp  []int
        var nextQ []*TreeNode
        
        for len(queue) != 0 {
            cur := queue[0]
            queue = queue[1:]
            temp = append(temp, cur.Val)
            if cur.Left != nil {
                nextQ = append(nextQ,cur.Left)
            }
            
            if cur.Right != nil {
                nextQ = append(nextQ,cur.Right)
            }
        }
        
        res = append(res, temp)
        queue = append(queue, nextQ...)        
    }
    
    return res
}
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2019年08月16日,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档