[路飞] leetcode 剑指 Offer 32 - II. 从上到下打印二叉树 II

Offer 驾到，掘友接招！我正在参与2022春招打卡活动，点击查看活动详情。

一、题目描述

leetcode 剑指 Offer 32 - II. 从上到下打印二叉树 II

从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。

例如:
给定二叉树: [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回其层次遍历结果：

[
  [3],
  [9,20],
  [15,7]
]

提示：

1. 节点总数 <= 1000

二、思路分析

层序打印二叉树，每层结果放到对应数组中, 每层从左到右顺序打印。
需要一个值记录当前节点应该放到哪个数组中

递归

无需展开理解，利用递归函数的语意信息去进行程序设计。
从上到下层序遍历，每层结果放到一个数组中，使用k记录节点放到哪一个数组中。
每层节点是从左到右依次存入数组中，所以先处理左子树，再处理右子树。

利用队列

先根节点root入队，root出队后，其左右子节点入队，然后依次，先看左侧，左子节点出队如果其有孩子们就将其左右孩子入队，同样的方式处理右子树

三、JavaScript代码

/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number[][]}
 */
// 递归
function getResult(root, k, ans){
    if(root == null) return root;
    if(k == ans.length) ans.push([]);
    ans[k].push(root.val);
    getResult(root.left, k+1, ans);
    getResult(root.right, k+1, ans);
}
var levelOrder = function(root) {
    let ans = []
    getResult(root, 0, ans)
    return ans
};


// 队列

var levelOrder = function(root) {
    if(root == null) return [];
    let queue = [];
    let ans = []
    queue.push(root)
    while(queue.length) {
        let arr = []
        for(let i = queue.length; i > 0; i --){
            let node = queue.shift()
            arr.push(node.val)
            if(node.left) queue.push(node.left);
            if(node.right) queue.push(node.right)
        }
        ans.push(arr);
    }
    return ans
};

四、总结