樹搜索 - 給定樹中的兩個節點，檢查它們是否連接 - python

Question

給定一個搜索樹，例如

"1"
 └ "2"
    ├ "2.1"
    ┊   └ "3"
    ┊
    └ "2.2"
        └ "2.2.1"
             └ "3"

以及屬於該樹的兩個節點a和b ，例如“2.1”和“3”。 我們如何檢查a和b是否是父子（或子父）相關/連接的？

對於第一個示例，應該生成 True。 這里還有一些：

a="3"      b="1"    -> False
a="3"      b="2"    -> False
a="2.2.1"  b="2.2"  -> True
a="2.2.1"  b="3"    -> True

我目前正在使用anytree庫，我正在努力實現這個解決方案。 上圖是結構上的簡化。 我目前嘗試實施的概述如下： https://pastebin.com/Mjk7gyqH

如果可以用純 python 或 anytree 給出答案，那就太好了，但任何答案總比沒有好。

Answer 1

您可以使用簡單的遞歸：

tree = {'name': '1', 'children': [{'name': '2', 'children': [{'name': '2.1', 'children': [{'name': '3'}]}, {'name': '2.2', 'children': [{'name': '2.2.1', 'children': [{'name': '3'}]}]}]}]}
def in_tree(d, node):
   return d['name'] == node or any(in_tree(i, node) for i in d.get('children', []))

def lookup(tree, a, b, flag=False):
   if tree['name'] == b and flag:
     return True
   return any(lookup(j, a, b, tree['name'] == a) for j in tree.get('children', []))

test = [['3', '1'], ['3', '2'], ['2.2.1', '2.2'], ['2.2.1', '3'], ['60', '70']]
for a, b in test:
   if not in_tree(tree, a) or not in_tree(tree, b):
      raise AttributeError('Node(s) missing in tree')
   print(any([lookup(tree, a, b), lookup(tree, b, a)]))

Output：

False
False
True
True
Traceback (most recent call last):
  File "<stdin>", line 3, in <module>
AttributeError: Node(s) missing in tree

Answer 2

如果我理解得很好，您“只是”要求沒有任何中間節點的直接父子關系。 如果這不是您要查找的內容，請提供另一個示例，顯示以下代碼失敗的位置，我可以修復它。

該代碼使用 anytree，因為這是您建議的庫

from anytree import Node, RenderTree

nodes = {} # a dict as a lookup to find nodes by name

def add_node(val, parentval=None):
    if parentval is not None:
        node = nodes[val] = Node(val, parent=nodes[parentval])
    else:
        node = nodes[val] = Node(val)
    return node

def mk_tree():
    top = add_node("1")
    add_node("2", "1")
    add_node("2.1", "2")
    add_node("3", "2.1")
    add_node("2.2", "2")
    add_node("2.2.1", "2.2")
    add_node("3", "2.2.1")
    return top

def is_child_or_parent(n1, n2):
    return n1.parent == n2 or n2.parent == n1

testpatterns = [
    ("3", "1", False),
    ("3", "2", False),
    ("2.2.1", "2.2", True),
    ("2.2.1", "3", True),
    ]

def run_test():
    for name1, name2, expected in testpatterns:
        node1 = nodes[name1]
        node2 = nodes[name2]
        rslt = is_child_or_parent(node1, node2)
        print(node1, node2, expected, rslt)
        assert rslt == expected

tree = mk_tree()
print(RenderTree(tree))
run_test()