找工作刷题记录_018二叉树的层次遍历

最新推荐文章于 2025-02-24 17:52:56 发布

原创最新推荐文章于 2025-02-24 17:52:56 发布 · 131 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#找工作刷题

本文介绍了一种算法，用于实现二叉树的层次遍历并返回每层节点的值。通过使用队列来跟踪每一层的节点，该算法能够依次访问所有节点，并将结果按层次组织。

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

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

返回其层次遍历结果：

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

# 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 = []
        queue = collections.deque()
        queue.append(root)
        
        #visited = set(root)
        
        while queue:
            level_size = len(queue)
            current_level = []
            
            for _ in range(level_size):
                node = queue.popleft()
                current_level.append(node.val)
                if node.left:queue.append(node.left)
                if node.right:queue.append(node.right)
                    
            result.append(current_level)
        return result

需要再理解。。。