查找根中是否有一对叶子路径，其总和等于根的数据

介绍

有时我们需要找到树中是否有一对叶子节点路径，其总和等于根节点的数据。这种情况在树的遍历、搜索等场景中比较常见。以下是一个实现该功能的伪代码：

# 定义一个函数来判断是否有一对叶子路径的和等于根节点的数据
def has_leaf_path_sum(root):
    # 用来存储从当前节点到每个叶子节点的路径
    paths = []
    # 获取每条从根节点到叶子节点的路径，并保存在 paths 数组中
    get_paths(root, [], paths)
    # 遍历 paths 数组，找到是否有一对路径满足其和等于根节点的数据
    for i in range(len(paths)):
        for j in range(i+1, len(paths)):
            if sum(paths[i]) + sum(paths[j]) == root.val:
                return True
    return False

# 用来获取从当前节点到每个叶子节点的路径
def get_paths(node, path, paths):
    # 如果当前节点是叶子节点，则将当前路径添加到 paths 数组中
    if not node.left and not node.right:
        path.append(node.val)
        paths.append(path)
        return
    # 否则继续递归获取路径
    if node.left:
        get_paths(node.left, path+[node.val], paths)
    if node.right:
        get_paths(node.right, path+[node.val], paths)

解释

上面的代码实现了一个名为 has_leaf_path_sum 的函数，该函数接收一个根节点，并返回一个布尔值表示是否存在一对叶子节点路径的和等于根节点的数据。

该函数首先调用 get_paths 函数来获取从根节点到每个叶子节点的路径，并将这些路径保存在 paths 数组中。然后通过两个嵌套循环遍历 paths 数组，找到是否有一对路径满足其和等于根节点的数据。如果找到了这样的一对路径，则说明树中存在一对叶子节点路径的和等于根节点的数据，返回 True，否则返回 False。

get_paths 函数用来递归获取从当前节点到每个叶子节点的路径，并将这些路径保存在 paths 数组中。具体来说，当当前节点是叶子节点时，将当前节点的值添加到路径中，并将路径添加到 paths 数组中。否则，依次递归左子树和右子树，并将当前节点的值添加到路径中。

示例

以下是该函数的使用示例：

# 构造一棵二叉树如下
#     3
#   /   \
#  2     5
# / \     \
#1   4     6
root = TreeNode(3)
root.left = TreeNode(2)
root.right = TreeNode(5)
root.left.left = TreeNode(1)
root.left.right = TreeNode(4)
root.right.right = TreeNode(6)
print(has_leaf_path_sum(root)) # 输出 True

总结

本文介绍了如何查找树中是否存在一对叶子节点路径的和等于根节点的数据。通过递归遍历树来获取每个叶子节点到根节点的路径，并遍历路径数组来查找是否存在一对路径满足条件。这是树的常见问题，在树的遍历、搜索等场景中非常有用。