LeetCode Algorithm 102. 二叉树的层序遍历

Posted 2022-03-28 Alex_996

102. 二叉树的层序遍历

Ideas

二叉树的层序遍历类似于图的广度优先搜索，都是要借助队列这种数据结构来实现。

首先将根节点添加到队列中，然后就是套BFS的模板，稍微改动一下。

只要队列不为空就循环：

1. 当前队列大小即为当前层元素个数；
2. 遍历当前层元素，依次取出队首元素，添加到返回结果中；
3. 将该元素的左右孩子节点添加到队列中

Code

Python

from typing import List
from collections import deque


# Definition for a binary tree node.
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 []

        res = []
        queue = deque()
        queue.append(root)
        while queue:
            res.append([])
            level_size = len(queue)
            for _ in range(level_size):
                node = queue.popleft()
                res[-1].append(node.val)
                if node.left:
                    queue.append(node.left)
                if node.right:
                    queue.append(node.right)
        return res