以下是一个按照父ID递归搜索后代的代码示例:
def search_descendants(nodes, parent_id):
descendants = []
for node in nodes:
if node['parent_id'] == parent_id:
descendants.append(node)
descendants.extend(search_descendants(nodes, node['id']))
return descendants
在这个示例中,nodes
是一个包含节点信息的列表,每个节点由一个字典表示,其中包含了节点的 id
和 parent_id
。parent_id
表示该节点的父节点的 id
。函数 search_descendants
的参数是节点列表和要搜索的父节点的 id
。
函数首先创建一个空列表 descendants
用于存储所有后代节点。然后遍历节点列表,对于每个节点,如果它的 parent_id
等于给定的父节点的 id
,则将该节点添加到 descendants
列表中,并递归调用 search_descendants
函数来搜索该节点的后代节点。最后,返回所有后代节点的列表。
以下是一个使用示例:
nodes = [
{'id': 1, 'parent_id': None},
{'id': 2, 'parent_id': 1},
{'id': 3, 'parent_id': 1},
{'id': 4, 'parent_id': 2},
{'id': 5, 'parent_id': 2},
{'id': 6, 'parent_id': 3},
{'id': 7, 'parent_id': 3},
{'id': 8, 'parent_id': 4},
{'id': 9, 'parent_id': 4},
]
descendants = search_descendants(nodes, 1)
print(descendants)
输出结果为:
[
{'id': 2, 'parent_id': 1},
{'id': 4, 'parent_id': 2},
{'id': 5, 'parent_id': 2},
{'id': 8, 'parent_id': 4},
{'id': 9, 'parent_id': 4},
{'id': 3, 'parent_id': 1},
{'id': 6, 'parent_id': 3},
{'id': 7, 'parent_id': 3}
]
这个示例中,给定的父节点的 id
是 1,函数会搜索父节点为 1 的所有后代节点,并将它们存储在列表 descendants
中返回。