102. 二叉树的层序遍历(python)

最新推荐文章于 2025-02-11 14:32:44 发布

原创最新推荐文章于 2025-02-11 14:32:44 发布 · 2.7k 阅读

7 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树 #算法 #leetcode #python #数据结构

本文介绍了一种实现二叉树层序遍历的方法，通过使用栈来存储每层的节点，从而实现了从上到下、从左到右的遍历顺序。

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

# 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]]:
        if not root:
            return
        result = []
        stack = [root]
        while stack:
            lay=[]
            lay_value=[]
            for node in stack:
                lay_value.append(node.val)
                if node.left:
                    lay.append(node.left)
                if node.right:
                    lay.append(node.right)
            stack = lay
            result.append(lay_value)
        return result