一棵圣诞树记作根节点为 root 的二叉树,节点值为该位置装饰彩灯的颜色编号。请按照从 左 到 右 的顺序返回每一层彩灯编号。

示例 1:

**输入:**root = [8,17,21,18,null,null,6]
输出:[8,17,21,18,6]

提示:

节点总数 <= 1000

层序遍历二叉树

借助队列的先入先出

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
class Solution {
public:
vector<int> decorateRecord(TreeNode *root) {
queue<TreeNode *> mid;
vector<int> ans;
if (root != nullptr)
mid.push(root);
while (!mid.empty()) {
TreeNode *cur = mid.front();
mid.pop();
if (cur->left != nullptr)
mid.push(cur->left);
if (cur->right != nullptr)
mid.push(cur->right);
ans.push_back(cur->val);
}
return ans;
}
};