Leetcode102. 二叉树的层序遍历 Binary Tree Level Order Traversal - Python 以递归/迭代算法实现

56 篇文章 1 订阅
47 篇文章 0 订阅
# 递归法
class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        result = []
        if not root: return []

        def traversal(root: TreeNode, depth = 0):
            if len(result) == depth: result.append([])  #当遍历到新的下一层后,新建一层[]
            result[depth].append(root.val) #遇到根节点就将其加入相应层中
            if root.left: traversal(root.left, depth + 1)
            if root.right: traversal(root.right, depth + 1) 
            #实际上该递归的结束条件为 root.left == None 和 root.right == None
        traversal(root, 0)
        return result

递归思路:

找准递归的三要素:

1.调用函数参数:root,depth-当前深度

2.函数体具体操作:当遇到新层,则添加[];遇到父节点就将root.val输出;然后分别将左孩子、右孩子依次用遍历父节点一样的方式(递归的方式)遍历孩子节点。

3.递归终止条件: #实际上该递归的结束条件为 root.left == None 和 root.right == None

需理解:

递归算法层序遍历二叉树,不是像迭代一样,依次遍历每一层的节点。而是深度优先,先依次遍历并到达最深的孩子节点,然后再往回走。如此,可能深层已经被遍历完毕,而浅层却还没有被遍历完毕。

class Solution:
    """二叉树层序遍历迭代解法"""

    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        results = []
        if not root: return results

        from collections import deque
        queue = deque([root])


        while queue:
            child_size = len(queue) #每一层待遍历的节点个数
            result = []    
            for _ in range(child_size):
                cur = queue.popleft()
                result.append(cur.val)

                #将左右孩子依次加入下一层待遍历节点队列中(dequeue)
                if cur.left: queue.append(cur.left)    
                if cur.right: queue.append(cur.right)
            results.append(result)
        return results

迭代法思路:

调用deque双向队列,double-ended queue;

将每一层的节点依次从右侧(默认)加入dequeue;

依次遍历每一层的节点值,对每一个节点,遇到左右孩子非空的,加入到dequeue中;

每一层的节点遍历结果记录在result中,每一层遍历结束后,将结果加入results。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值