如果帮助到您,还请点个关注吧,hahaha
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
例如:
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其自底向上的层序遍历为:
[
[3],
[9,20],
[15,7]
]
#最直观的思路
使用两种方法来解题:
1、DFS 2、BFS
DFS思路:
判断当前节点所处的高度,如果ans的长度小于高度值,
则在ans里新建一个list用来存放新高度中的数字。
存放完之后判断当前节点是否含有子节点,如有则进行递归操作。
依次进行便可求出结果。
BFS思路:
利用deque包可以直接popleft大大降低复杂度
nowlen用来表示当前高度含有几个数值,
随后利用for循环来依次存进去在同一个nowlen上的值
同时检查当前节点是否存在子节点,
若存在,则将子节点存放到queue中当做下一高度上的数字。
依次类推,直至等到所有节点都没有子节点
Ps:巧妙利用deque包,可以逐步输出经deque操作后的结果,便于更好的理解代码
from collections import deque
class Solution:
def levelOrder(self, root: TreeNode) -> List[List[int]]:
if root == None:
return []
ans = []
'''
def Node(root, lev):
if len(ans) < lev + 1:
ans.append([])
ans[lev].append(root.val)
if root.left:
Node(root.left, lev + 1)
if root.right:
Node(root.right, lev + 1)
Node(root, 0)
return ans
'''
queue = deque([root])
while queue:
nowlen = len(queue)
now_val = []
for _ in range(nowlen):
node = queue.popleft()
now_val.append(node.val)
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
ans.append(now_val)
return ans