剑指offer34：二叉树中和为某一值的路径

最新推荐文章于 2023-06-21 19:29:26 发布

原创最新推荐文章于 2023-06-21 19:29:26 发布 · 194 阅读

0 ·

CC 4.0 BY-SA版权

剑指offer 专栏收录该内容

12 篇文章

订阅专栏

本文介绍了一种在二叉树中寻找所有达到特定数值路径的算法。通过递归方式，从根节点开始，减去节点值并向下递归左子树和右子树，直到叶子节点，检查路径总和是否等于期望值。如果相等，则将该路径加入结果列表。

在这里插入图片描述
1.判断是不是空树
2.递归结束的条件是叶子节点
3.考察二叉树前序遍历

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    # 返回二维列表，内部每个列表表示找到的路径
    def FindPath(self, root, expectNumber):
        # write code here
        if not root:
            return []
        if root and not root.left and not root.right and root.val == expectNumber:
            return [[root.val]]
        res = []
        left = self.FindPath(root.left, expectNumber-root.val)
        right = self.FindPath(root.right, expectNumber-root.val)
        for i in left+right:
            res.append([root.val]+i)
        return res