PeterQwQ
0%

Leetcode 102 二叉树的层序遍历

发表于

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

就是BFS，用队列存储每一层的节点，然后遍历队列，每次遍历一层，然后将下一层的节点加入队列。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
# 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: Optional[TreeNode]) -> List[List[int]]:
        if not root:
            return []
        queue, cur_level_ans,ans = [root], [], []
        while queue:
            tmp = []
            for node in queue:                
                if node.left:
                    tmp.append(node.left)
                if node.right:
                    tmp.append(node.right)           
                cur_level_ans.append(node.val)
            queue = tmp
            ans.append(cur_level_ans)
            cur_level_ans = []
        return ans