⚡️102. 二叉树的层序遍历⚡️
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例:
二叉树:[3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层序遍历结果:
[
[3],
[9,20],
[15,7]
]
⚡️分析⚡️
层序遍历用广度优先遍历算法即可,队列存储每层压入的结点。
整体思路是:先创立一个要返回的数组res,在压入每层结点时,先对res压入一个空数组,然后对于该空数组,通过队列的长度来进行相应次数的while循环,来压入结点且继续对左右结点进行压入队列
例如第二层是9,20,往res压入一个空数组后,进行两次while循环,然后往空数组依次压入9,20,然后每次压入时也分别对9,20的左右结点进行压入队列,以便第三个空数组时压入
对于while循环,用队列的长度来控制
对于空数组,用res[res.length-1]来控制,刚好压到res数组res.length-1下标的空数组下。
代码如下:
/**
* 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) {
if(!root)return [];
const q = [root];
const res = [];
while(q.length){
let len = q.length;
res.push([]);
while(len--){
const p = q.shift();
res[res.length-1].push(p.val);
if(p.left) q.push(p.left);
if(p.right) q.push(p.right);
}
}
return res;
};
算法效率如图:
时间复杂度和空间复杂度依旧为O(n)
觉得该篇文章有用的请不要忘记忘记点击右下角的大拇指~
欢迎大家关注我的公众号:Smooth前端成长记录
公众号正在努力更新CSDN博客内容,想方便阅读博客的C友可以来关注我的公众号以便获得更优良的阅读体验~