题目描述
给你二叉树的根节点 root
,返回其节点值的 层序遍历 。 (即逐层地,从左到右访问所有节点)。
示例 1:
输入: root = [3,9,20,null,null,15,7]
输出: [[3],[9,20],[15,7]]
示例 2:
输入: root = [1]
输出: [[1]]
示例 3:
输入: root = []
输出: []
提示:
- 树中节点数目在范围
[0, 2000]
内 -1000 <= Node.val <= 1000
分析解答
从输出入口分析我们需要做什么。输出为一个二维数组,所以需要一个 res 作为最终结果数组,需要一个个的 temp 作为每一层的数组。
那么如何控制元素的遍历条件呢?使用队列!
队列中只要有元素,就需要循环遍历。为了将当前层的元素添加到 temp 当前层的数组中。而循环的次数刚好就是当前 queue 的长度(当前层元素个数)。将旧的(当前层元素)推入 temp,将新的(子节点)添加到 queue。
注:需要先保存一下 length 因为 在获取子节点的时候会改变 queue.length
js
/**
* Definition for a binary tree node.
* function TreeNode(val, left, right) {
* this.val = (val===undefined ? 0 : val)
* this.left = (left===undefined ? null : left)
* this.right = (right===undefined ? null : right)
* }
*/
/**
* @param {TreeNode} root
* @return {number[][]}
*/
var levelOrder = function (root) {
let res = []
let queue = []
if (!root) {
return res
}
queue.push(root);
while (queue.length) {
// 当前层的节点
let temp = []
// 需要先保存一下 length 因为 在获取子节点的时候会改变 queue.length
let length = queue.length
for (let i = 0; i < length; i++) {
let node = queue.shift()
temp.push(node.val)
node.left && queue.push(node.left)
node.right && queue.push(node.right)
}
res.push(temp)
}
return res
};
上图来自:代码随想录