Question

我的树具有以下结构： tree = {'0':('1'，'2'，'3'），'1':('4'），'2':('5'，'6'），'3':( ），'4':('7'，'8'），'8':('9'，'10'，'11'）}

如何编写Python代码来检索特定节点的所有给定子节点？例如，如果我给它节点4，则代码应检索7,8,9,10,11。对于节点2，它应该检索5、6，依此类推。

我刚刚开始学习Python的基础知识，但我不知道如何为非二进制树实现这一点。

Answer 1

您可以使用队列。

获取用户请求的值后，将其推入队列。然后，在队列不为空的情况下，弹出一个值，进行打印，检查dict，如果当前值是dict中的键，则将每个这些值添加到队列中以进行检查在下一遍。

import queue

tree={'0':('1','2','3'), '1':('4'), '2':('5','6'), '3':(), '4':('7','8'), '8':('9','10','11')}

num = input("what you want ")

q = queue.Queue()

q.put(num)

while not q.empty():
  n = q.get()
  for s in n:
    print(s)
    if s in tree:
      q.put(tree[s])

Demo

请注意，如果您有树tree={'0':('1'), '1':('0')}或任何其他循环引用，则此代码将永远运行。小心！

如何在Python中获取通用树的所有子节点

1 个答案:

您可以使用队列。