在python中提取树的根

Question

嘿大家， 我一直在尝试找到一个用于在python中提取树的根的内置函数， 我没有找到类似的东西，我一直在努力构建我自己的东西，但我无法构建一些通用的东西以满足我的所有需求。 有没有人准备好或者知道如何从python中的树结构中提取这些信息？

谢谢

Answer 1

你必须自己动手：

class Node(object):
    def __init__(self, p=None):
        self.parent = p
        self.children = []

n1 = Node()
n2 = Node()
n1.children.append(n2)
n2.parent = n1

当然，您希望使用addChild等方法自动管理所涉及对象的.children和.parent属性。

然后你可以写一个方法

def findRoot(node):
    p = node
    while p.parent != None:
        p = p.parent
    return p