LeetCode刷题之路:102. 二叉树的层序遍历

如果帮助到您,还请点个关注吧,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

猜你喜欢

转载自blog.csdn.net/as812252319/article/details/112442706