我对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
我的要求是使用级别顺序遍历在列表中获取树的元素。
答案 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