使用dict vs queue

时间:2016-11-18 10:15:02

标签: python time-complexity tree-traversal breadth-first-search

我对Algo& amp;的理解DS有点新手。而且我不确定这是一个重复或相关的问题,还是完全无关紧要的问题。无论我在哪里看到级别订单遍历或BFS被提及,我都会看到使用队列。我无法理解其中的复杂性,在空间方面,更重要的是时间复杂度反对使用词典进行实施。

def getLevelElements(tree, level=0, cont={}):
    """Get mapping of level and elements in a level
    :param tree: Input tree, ex.
        1
            2
                4
                5
                    8
                    9
            3
                6
                7
                    10
                    None
    :return: {0: [1], 1: [2, 3], 2: [4, 5, 6, 7], 3: [8, 9, 10]}
    """
    if tree is not None:
        cont.setdefault(level, []).append(tree.root)
    if tree.leftChild is not None:
        getLevelElements(tree.leftChild, level + 1, cont)
    if tree.rightChild is not None:
        getLevelElements(tree.rightChild, level + 1, cont)
    return cont


def levelOrderTraversalOut(tree):
    levelElementsMap = getLevelElements(tree)
    out = []
    for elements in levelElementsMap.values():
        out += elements
    return out

我的要求是使用级别顺序遍历在列表中获取树的元素。

1 个答案:

答案 0 :(得分:3)

如果我理解你,你需要树的元素的BFS顺序。 我会建议像这样的人:

def bfs(tree):
    out = []
    elements = [tree.root]
    while elements:
        elem = elements.pop(0)
        out.append(elem)
        if elem.leftChild:
            elements.append(elem.leftChild)
        if elem.rightChild:
            elements.append(elem.rightChild)
    return out