102. 二叉树的层序遍历
102. 二叉树的层序遍历
题解
简单的队列,比较考验语法熟练度而已。。
注意27行那里有个坑,应该是queue[0],而不是queue[i],因为每次都出队了,下一个索引位置肯定还是0
代码
package main
func main() {
}
type TreeNode struct {
Val int
Left *TreeNode
Right *TreeNode
}
func levelOrder(root *TreeNode) [][]int {
if root == nil {
return nil
}
result := make([][]int, 0)
queue := make([]*TreeNode, 0)
queue = append(queue, root)
for len(queue) > 0 {
cnt := make([]int, 0)
queueSize := len(queue)
for i := 0; i < queueSize; i++ {
//出队
v := queue[0]
queue = queue[1:]
//入队
if v.Left != nil {
queue = append(queue, v.Left)
}
if v.Right != nil {
queue = append(queue, v.Right)
}
cnt = append(cnt, v.Val)
}
result = append(result, cnt)
}
return result
}