级别订单遍历BST代码

时间:2018-05-31 02:02:23

标签: python binary-search-tree

我想知道是否有人可以在BST中查看我的代码以进行级别订单遍历。我想有[[3],[9,20],[15,7]]但是得到[[3],[9],[20],[15],[7]]我知道我需要一个外部while循环,但是不知道如何构造它。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

import queue
class Solution(object):
    def levelOrder(self, root):
        L = queue.Queue()
        local = []
        L.put(root)
        while not L.empty():
            node = L.get()
            local.append([node.val])
            if(node.left):
                L.put(node.left)
            if(node.right):
                L.put(node.right)
        return local

0 个答案:

没有答案