二叉树的层序遍历 LCR 149.彩灯装饰记录-i

源代码: https://leetcode.cn/problems/cong-shang-dao-xia-da-yin-er-cha-shu-lcof/description/

/*
 * @lc app=leetcode.cn id=LCR 149 lang=golang
 * @lcpr version=30204
 *
 * [LCR 149] 彩灯装饰记录 I
 */

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func decorateRecord(root *TreeNode) []int {
	if root == nil {
		return []int{}
	}
	ans := []int{}

	queue := []*TreeNode{root}
	for len(queue) > 0 {
		for i := 0; i < len(queue); i++ {
			node := queue[0]  // 取第一个元素
			queue = queue[1:] // 删除第一个元素
			ans = append(ans, node.Val)
			if node.Left != nil {
				queue = append(queue, node.Left)
			}
			if node.Right != nil {
				queue = append(queue, node.Right)
			}
		}
	}
	return ans
}