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

104 阅读1分钟

题目介绍

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

 

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

3

/
9 20 /
15 7 返回其层次遍历结果:

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

提示:

节点总数 <= 1000 注意:本题与主站 102 题相同:leetcode-cn.com/problems/bi…

来源:力扣(LeetCode) 链接:leetcode-cn.com/problems/co… 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解题思路

递归法, 定义一个二维数组存储返回值,每次判断当前层级是否有对应的数组,没有就初始化数组并压入当前值,然后左右子树递归调用该方法,并把层级传进去

代码

/**
 * 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) {
    let arr = []
    function levelOrderFn(root, i = 0) {
        if (!root) return
        if (!arr[i]) arr[i] = []
        arr[i].push(root.val)
        levelOrderFn(root.left, i + 1)
        levelOrderFn(root.right, i + 1)
    }
    levelOrderFn(root)
    return arr
};