Leetcode日记102. 二叉树的层序遍历给你二叉树的根节点 root ,返回其节点值的 层序遍历
给你二叉树的根节点 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
- 按理来说二叉树的各种遍历应该是我们的基本功,可也正因为这样,我们在遇到这些经典问题时,最先就开始找别人或是自己曾经的记录,而忘记了自己的基本功。
- 我看到,题目的输出样例他把每一层都放到了一个列表内,这让我走进了误区,我才用了一个标记0,来在队列之中隔开每一曾,而这也导致了我的执行速度慢了下来。
- 首先建一个队列用于暂存当前层的非空子节点,再用while来进行遍历节点,遇到层尾标记0且不是最后一层,就添加新层的列表。遇到节点就依次遍历其左右节点,把非空节点添加到queue队列之中,并输出该节点到return列表中。直到最后只有一个标记0时返回层序遍历二叉树结果。
if root is None:
return []
queue = [root,0]
queue_len = 2
result = [[]]
while queue :
if queue[0] != 0 :
result[-1].append(queue[0].val)
if queue[0].left :
queue.append(queue[0].left)
queue_len += 1
if queue[0].right :
queue.append(queue[0].right)
queue_len += 1
queue_len -= 1
queue.pop(0)
elif queue[0] == 0 and queue_len > 1:
result.append([])
queue_len -= 1
queue.pop(0)
queue.append(0)
queue_len += 1
else :
return result
- 这里就可以看到这次的执行速度相对而言没有那么快,这就是因为我的标记位0所花费的时间。
- 这个是找到的比较好的结果,它采用了双层循环,但时间复杂度仍为n,因为仍旧是每个节点遍历一次。它使用tmp列表来暂存每一层的节点,第二次for循环之前清空,而不是像我一样使用标记位这节省了一部分的时间。
if not root:
return []
queue = [root]
res = []
while queue:
tmp = []
size = len(queue)
for i in range(size):
node = queue.pop(0)
tmp.append(node.val)
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
res.append(tmp)
return res