一棵圣诞树记作根节点为 root
的二叉树,节点值为该位置装饰彩灯的颜色编号。请按照从左到右的顺序返回每一层彩灯编号,每一层的结果记录于一行。
示例 1:
输入:root = [8,17,21,18,null,null,6]
输出:[[8],[17,21],[18,6]]
提示:
节点总数 <= 1000
思路: 二叉树的层次遍历、广度优先遍历
1、如果根结点为空,直接返回{}。
2、创建队列q并初始化将根结点入队(利用先进先出特性)。
3、广度优先循环(队列不为空)
计算当前队列长度size,创建临时数组。
(在循环体外部创建临时数组会导致开始遍历当前层次时临时数组不为空)
根据当前队列长度size大小遍历当前层次结点。
1.队首元素插入临时数组尾部。
2.依次将队首元素不为空的左子树、右子树入队。
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<vector<int>> decorateRecord(TreeNode* root) {
if(!root)
return {};
queue<TreeNode*>q;
vector<vector<int>>res;
q.push(root);
while(!q.empty())
{
int size = q.size();
vector<int>tmp;
for(int i = 0; i < size; ++i) //以每一层为单位
{
tmp.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();
}
res.push_back(tmp);
}
return res;
}
};