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

​​102. 二叉树的层序遍历​​

Ideas

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

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

只要队列不为空就循环：

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