二叉树的层序遍历
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例: 二叉树:[3,9,20,null,null,15,7],
3 / \ 9 20 / \ 15 7
返回其层次遍历结果:
[ [3], [9,20], [15,7] ]
解题思路
二叉树的层序遍历,即二叉树的广度遍历,先遍历根节点的相邻节点,再一次遍历相邻节点的子节点。广度遍历通常借助队列来实现。用队列来存储当前层的节点数,遍历当前层的节点,将当前层的节点依次推入数组subRes[],再将当前节点的左右子节点推入队列中,进入下一层进行遍历,直到遍历完整棵树,即完成到二叉树的层序遍历。
/**
* Definition for a binary tree node.
* function TreeNode(val) {
* this.val = val;
* this.left = this.right = null;
* }
*/
/**
* @param {TreeNode} root
* @return {number[][]}
*/
//广度遍历 + 队列
var levelOrder = function(root) {
if(!root){
return []
}
var res = [] //结果最外层数组
var queue = [root]
while(queue.length>0){
var len = queue.length//当前层的节点数目
var subRes = []
for(var i=0;i<len;i++){
var node = queue.shift() //节点出列
subRes.push(node.val) //将当前层的节点值加入subRes数组中
//将下一层节点计入队列中
if(node.left){
queue.push(node.left)
}
if(node.right){
queue.push(node.right)
}
}
res.push(subRes)
}
return res
};