文章

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 进行授权