「这是我参与2022首次更文挑战的第6天,活动详情查看:2022首次更文挑战」。
题目
给你二叉树的根节点 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
思路
一眼就能看出是bfs,但是这个bfs怎么写呢?
什么是层序遍历呢?简单来说,层序遍历就是把二叉树分层,然后每一层从左到右遍历:
如果我们使用 DFS/BFS 只是为了遍历一棵树、一张图上的所有结点的话,那么 DFS 和 BFS 的能力没什么差别,我们当然更倾向于更方便写、空间复杂度更低的 DFS 遍历。不过,某些使用场景是 DFS 做不到的,只能使用 BFS 遍历
首先判断root是否为空节点 之后先将root添加到队列中,此时队列的头为根节点, 之后进行循环,条件为quene不为空队列,然后记录每一层节点的个数,对这一层的每个节点遍历
代码
/**
* Definition for a binary tree node.
* function TreeNode(val, left, right) {
* this.val = (val===undefined ? 0 : val)
* this.left = (left===undefined ? null : left)
* this.right = (right===undefined ? null : right)
* }
*/
/**
* @param {TreeNode} root
* @return {number[][]}
*/
var levelOrder = function (root) {
let quene = [];//队列
let res = [];//结果数组
if (!root) {
return res
}
quene.push(root)//现将根节点添加到队列中
res.push([root.val])
while (quene.length) {
let temp = [];
let num = quene.length//记录每一层的节点个数
while (num--) {//对每层的每个节点进行遍历,找到左右子节点,如果有就添加到临时数组temp中
let cur = quene.pop()
if (cur.left) {
quene.unshift(cur.left)
temp.push(cur.left.val)
}
if (cur.right) {
quene.unshift(cur.right)
temp.push(cur.right.val)
}
}
if (temp.length)//如果temp不为空 就把temp添加到res数组中
res.push(temp)
}
return res
};
时间复杂度:假设有n层的话每层m个元素时间复杂度就是O(mn)