题目
给你二叉树的根节点 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
};
思路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
};