我们可以使用深度优先搜索(DFS)来遍历树,并记录经过的路径。如果当前节点是目标节点,则将当前路径添加到结果列表中。否则,我们继续递归地遍历子节点,直到找到目标节点或者遍历所有可能路径。
以下是使用 Python 语言的代码示例:
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
def find_paths(root: TreeNode, target: int) -> List[List[int]]:
def dfs(node: TreeNode, path: List[int], res: List[List[int]]):
if not node:
return
path.append(node.val)
if node.val == target:
# 当前节点是目标节点,将路径加入结果列表
res.append(path[:])
dfs(node.left, path, res)
dfs(node.right, path, res)
# 回溯
path.pop()
res = []
dfs(root, [], res)
return res
在这个代码中,我们定义了一个内部的 dfs 函数,接受当前节点、当前路径和结果列表作为参数。在函数中,将当前节点的值添加到路径中,并进行如下操作:
最后,我们在外部函数中调用 dfs 函数,并返回结果列表。
使用如下代码调用函数:
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)
root.right.left = TreeNode(6)
root.right.right = TreeNode(7)
res = find_paths(root, 5)
print(res) # 输出 [[1, 2, 5]]
输出结果为 [[1, 2, 5]],表示根节点到目标节点 5 的路径为 [1, 2, 5]。
下一篇:遍历一列