leetcode刷题笔记-二叉树4

最新推荐文章于 2022-01-29 01:11:27 发布

弥钥

最新推荐文章于 2022-01-29 01:11:27 发布

阅读量316

点赞数

文章标签： leetcode 算法

本文链接：https://blog.csdn.net/qq_43509788/article/details/105470672

版权

本文详细介绍了二叉树的层序、前序、后序和中序遍历的递归与迭代解法，包括代码实现和思路解析，帮助理解二叉树遍历的不同方法。

摘要由CSDN通过智能技术生成

二叉树的层序遍历

给你一个二叉树，请你返回其按层序遍历得到的节点值。（即逐层地，从左到右访问所有节点）。

[
  [3],
  [9,20],
  [15,7]
]

今天又是抄代码的一天呢~。
递归真的是一看就会，一写就废，而且很多时候重复计算很多，我想试试看迭代了。
这道题中递归算法的本质就是判断len(res)==depth，如果成立，就新增加一个子列表，这样来完成层级的区分。然后再分别加入每层的值：

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        res = []
        
        def helper(root,depth):
            if not root:
                return 0
            if len(res)==depth:
                res.append([])
            res[depth].append(root.val)

            helper(root.left,depth+1)
            helper(root.right,depth+1)

            
        helper(root,0)
        return res