LeetCode热题(JS版) - 102. 二叉树的层序遍历

83 阅读1分钟

题目

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

示例 1:

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

示例 2:

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

示例 3:

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

提示:

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

思路1:队列

  • 宽度优先搜索(BFS)
  • 队列是先进先出(FIFO)
/**
 * Definition for a binary tree node.
 * class TreeNode {
 *     val: number
 *     left: TreeNode | null
 *     right: TreeNode | null
 *     constructor(val?: number, left?: TreeNode | null, right?: TreeNode | null) {
 *         this.val = (val===undefined ? 0 : val)
 *         this.left = (left===undefined ? null : left)
 *         this.right = (right===undefined ? null : right)
 *     }
 * }
 */

function levelOrder(root: TreeNode | null): number[][] {
    const queue = [];
    const result = [];
    if(!root) return result
    queue.push(root);
    while(queue.length) {
        let length = queue.length;
        const temp = [];
        while(length--) {
            const item = queue.shift();
            temp.push(item.val);
            item.left && queue.push(item.left);
            item.right && queue.push(item.right);
        }
        result.push(temp);
    }

    return result
};

image.png

思路2: 递归

  • 深度递归搜索(DFS)
/**
 * Definition for a binary tree node.
 * class TreeNode {
 *     val: number
 *     left: TreeNode | null
 *     right: TreeNode | null
 *     constructor(val?: number, left?: TreeNode | null, right?: TreeNode | null) {
 *         this.val = (val===undefined ? 0 : val)
 *         this.left = (left===undefined ? null : left)
 *         this.right = (right===undefined ? null : right)
 *     }
 * }
 */

function levelOrder(root: TreeNode | null): number[][] {
    const result = [];
    // BFS
    // const queue = [];
    // if(!root) return result
    // queue.push(root);
    // while(queue.length) {
    //     let length = queue.length;
    //     const temp = [];
    //     while(length--) {
    //         const item = queue.shift();
    //         temp.push(item.val);
    //         item.left && queue.push(item.left);
    //         item.right && queue.push(item.right);
    //     }
    //     result.push(temp);
    // }

    const DFS = (node, depth) => {
        if(!node) return
        if(!result[depth]) result[depth] = [];
        
        result[depth].push(node.val);

        node.left && DFS(node.left, depth + 1);
        node.right && DFS(node.right, depth + 1);
    }

    DFS(root, 0);
    return result
};

image.png

参考: www.cnblogs.com/tqlin/p/118…