leetcode102. 二叉树的层序遍历

题目描述

给你二叉树的根节点 root ，返回其节点值的 层序遍历 。 （即逐层地，从左到右访问所有节点）。

示例1:

输入： root = [3,9,20,null,null,15,7]
输出： [[3],[9,20],[15,7]]

思路

层序遍历:就是从左到右一层一层的去遍历。这种遍历的方式和我们之前接触过的都不太一样。采用二叉树的数据结构来做肯定不行，因为二叉树里面只能获取当前节点和当前节点的左右子节点。那我们应该怎样去做呢？

这里我们需要借用一个辅助数据结构即队列来实现：队列先进先出，符合一层一层遍历的逻辑，而用栈先进后出适合模拟深度优先遍历也就是递归的逻辑。

图片以及思路来自代码随想录层序遍历

代码

理清楚了思路，下面便可以撸代码了。

/**
 * 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) {
    // 初始化结果数组：result
    // 初始化队列：queue
    let result=[],queue=[];
    if(root)queue.push(root);

    while(queue.length){
        //队列长度
        let size=queue.length;
        //每层的数组
        let temp=[];

        //每层循环
        while(size--){
            //队首节点出队
            let node=queue.shift();
            //将结果放入temp中
            temp.push(node.val);
            // 将节点的左右子节点放入
            if(node.left) queue.push(node.left);
            if(node.right) queue.push(node.right);
            //判断该层是否循环完成
            if(size===0) result.push(temp);
        }
    }

    // 返回结果
    return result;
};

总结

二叉树的层序遍历的重点思路是要想到使用队列来进行一层一层的遍历。 今天的代码题：二叉树的层序遍历便做完了，如有错误之处，欢迎各位小伙留言指出，谢谢大家啦！