一棵圣诞树记作根节点为 root
的二叉树,节点值为该位置装饰彩灯的颜色编号。请按照从 左 到 右 的顺序返回每一层彩灯编号。
示例 1:
输入:root = [8,17,21,18,null,null,6]
输出:[8,17,21,18,6]
提示:
节点总数 <= 1000
思路:二叉树的层次遍历、广度优先遍历。
1、如果根结点为空,直接返回{}。
2、创建队列q并初始化将根结点入队(利用先进先出特性)。
3、广度优先循环(队列不为空)
将队首元素插入结果数组尾部。 依次将队首元素不为空的左子树、右子树入队。 队首元素出队。 4、返回结果数组。
代码实现:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
vector<int> decorateRecord(TreeNode* root) {
if(!root)
return {};
queue<TreeNode*>q;
vector<int>res;
TreeNode* tmp;
q.push(root);
while(!q.empty())
{
res.push_back(q.front()->val);
if(q.front()->left)
q.push(q.front()->left);
if(q.front()->right)
q.push(q.front()->right);
q.pop();
}
return res;
}
};