题目
从上到下打印出二叉树的每个节点,同一层的节点按照从左到右的顺序打印。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/
9 20
/
15 7
返回:
[3,9,20,15,7]
思路
//用queue存储当前遍历的一层所有节点的值,通过长度是否为0判断这一层是否有值
代码
/**
* 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 == null){
return [];
}
return bfs(root);
};
function bfs(root){
const queue = [root], result = [];
while(queue.length){
for(let i = 0, len = queue.length; i < len; i++){
let cur = queue.shift();
result.push(cur.val);
if(cur.left){
queue.push(cur.left);
}
if(cur.right){
queue.push(cur.right);
}
}
}
return result;
}