102-binary-tree-level-order-traversal
102-binary-tree-level-order-traversal
Given the root of a binary tree, return the level order traversal of its nodes’ values. (i.e., from left to right, level by level).
Example 1:
Input: root = [3,9,20,null,null,15,7] Output: [[3],[9,20],[15,7]] Example 2:
Input: root = [1] Output: [[1]] Example 3:
Input: root = [] Output: []
My solution:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution(object):
def levelOrder(self, root):
"""
:type root: Optional[TreeNode]
:rtype: List[List[int]]
"""
if not root:
return []
stack = [root]
res = []
while stack:
tmp = []
val = []
while stack:
cur = stack.pop(0)
val.append(cur.val)
if cur.left:
tmp.append(cur.left)
if cur.right:
tmp.append(cur.right)
stack = tmp[:]
res.append(val)
return res
本文由作者按照 CC BY 4.0 进行授权