LeetCode102 && LeetCode107:二叉树的层次遍历

目录

一、题目

二、示例

三、思路

四、代码


一、题目

1、LeetCode102:给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。

2、LeetCode107:给定一个二叉树,返回其节点值自底向上的层次遍历。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历

二、示例

1、LeetCode102:

示例:
二叉树:[3,9,20,null,null,15,7],

    3
   /  \
  9  20
      /  \
   15   7
返回其层次遍历结果:

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

2、LeetCode107:

示例:
给定二叉树 [3,9,20,null,null,15,7],

     3
    /  \
  9  20
      /  \
    15   7
返回其自底向上的层次遍历为:

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

三、思路

1、LeetCode102:先序遍历的思想,depth变量用来记录遍历的层数,每向下一层递归,depth变量+1

2、LeetCode107:同LeetCode102,不同的是,最后倒序的方式输出res中的值

四、代码

1、LeetCode102:

# 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):
        res = []

        def DFS(root, depth, res):
            if not root:
                return
            if depth >= len(res):
                res.append([])
            res[depth].append(root.val)
            DFS(root.left, depth + 1, res)
            DFS(root.right, depth + 1, res)

        DFS(root, 0, res)
        return res

if __name__ == '__main__':
    root = TreeNode(3)
    root.left = TreeNode(9)
    root.right = TreeNode(20)
    root.right.left = TreeNode(15)
    root.right.right = TreeNode(7)
    s = Solution()
    ans = s.levelOrder(root)
    print(ans)

2、LeetCode107:

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

class Solution:
    def levelOrderBottom(self, root: TreeNode):
        res = []

        def DFS(root, depth, res):
            if not root:
                return
            if depth >= len(res):
                res.append([])
            res[depth].append(root.val)
            DFS(root.left, depth + 1, res)
            DFS(root.right, depth + 1, res)

        DFS(root, 0, res)
        return res[::-1]

if __name__ == '__main__':
    root = TreeNode(3)
    root.left = TreeNode(9)
    root.right = TreeNode(20)
    root.right.left = TreeNode(15)
    root.right.right = TreeNode(7)
    s = Solution()
    ans = s.levelOrderBottom(root)
    print(ans)

猜你喜欢

转载自blog.csdn.net/weixin_45666660/article/details/108738276