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

Question

我的树结构如下：树={'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

您可以使用队列。

获得用户请求的值后，将其推入队列。然后，当队列不为空时，弹出一个值，打印它，检查字典，如果当前值是字典中的键，则将每个 those 值添加到排队等待下一次检查。

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 中通用树的所有子节点

How can I get all child nodes of general tree in Python

python

tree

child-nodes

您可以使用队列。