题目链接:https://leetcode.cn/problems/binary-tree-level-order-traversal/
难度:中等
描述:
给你二叉树的根节点 root
,返回其节点值的 层序遍历 。 (即逐层地,从左到右访问所有节点)。
提示:
节点数目:[0, 2000]
题解
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]]:
q = deque()
if root:
q.append(root)
ret = []
while q:
size = len(q)
temp = []
while size > 0:
node = q.popleft()
temp.append(node.val)
if node.left:
q.append(node.left)
if node.right:
q.append(node.right)
size -= 1
ret.append(temp)
return ret