每日一题--二叉树的锯齿形层序遍历

90 阅读2分钟

「这是我参与2022首次更文挑战的第7天,活动详情查看:2022首次更文挑战」。


题目

给你二叉树的根节点 root ,返回其节点值的 锯齿形层序遍历 。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。

 

示例 1:

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

示例 2:

输入:root = [1] 输出:[[1]]

示例 3:

输入:root = [] 输出:[]  

提示:

树中节点数目在范围 [0, 2000] 内 -100 <= Node.val <= 100


思路: 一眼就能看出是bfs,但是这个bfs怎么写呢?

什么是层序遍历呢?简单来说,层序遍历就是把二叉树分层,然后每一层从左到右遍历:

如果我们使用 DFS/BFS 只是为了遍历一棵树、一张图上的所有结点的话,那么 DFS 和 BFS 的能力没什么差别,我们当然更倾向于更方便写、空间复杂度更低的 DFS 遍历。不过,某些使用场景是 DFS 做不到的,只能使用 BFS 遍历

因为上面每次都是从左边开始打印,但这题要求的是先从左边,下一层从右边,在下一次从左边……左右交替的。我们可以使用一个变量,如果是true就表示从左边开始打印,否则就从右边开始打印,只需要把上面代码修改下即可,来看下代码:


代码


 * @param {TreeNode} root
 * @return {number[][]}
 */
var levelOrder = function (root) {
    let quene = [];//队列
    let res = [];//结果数组
    let n = 1;
    if (!root) {
        return res
    }
    quene.push(root)//现将根节点添加到队列中
    res.push([root.val])
    while (quene.length) {
        let temp = [];
        let num = quene.length//记录每一层的节点个数
        console.log(num)
        while (num--) {//对每层的每个节点进行遍历,找到左右子节点,如果有就添加到临时数组temp中
            let cur = quene.shift()
            if (cur.left) {
                quene.push(cur.left);
                temp.push(cur.left.val);
            }
            if (cur.right) {
                quene.push(cur.right);
                temp.push(cur.right.val)

            }
        }
        if (temp.length)//如果temp不为空 就把temp添加到res数组中
            res.push(temp)
    }
    return res
};