leetcode_102 二叉树的层序遍历

要求

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

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

    3
   / \
  9  20
    /  \
   15   7

返回其层序遍历结果：

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

代码详解

class TreeNode:
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right
        
class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        if not root:
            return []
        
        nodes = [root]
        result = list(list())
        self.generate(nodes,result)
        return result

    def generate(self,nodes,result):
        current_layer_values = []
        next_layer_node = []

        for node in nodes:
            if node.left:
                next_layer_node.append(node.left)
            if node.right:
                next_layer_node.append(node.right)
            current_layer_values.append(node.val)
        
        result.append(current_layer_values)

        if len(next_layer_node) == 0:
            return 
        
        self.generate(next_layer_node,result)

重点题目

解题思路： 这道题主要考察的点是树的层次遍历，核心是要构建两个层，第一个层是当前层，用于存储当前的数据值，另外的一层是用来存储下一层的节点，这样我们可以递归进行向下层次的搜索，当我们搜索的节点既没有左节点和右节点的时候，说明没有了下一层，这个时候，就是我们递归的出口，上述步骤即可完成整个的层次遍历。