【python-leetcode113-树的深度遍历】路径总和Ⅱ
发布日期:2021-05-09 08:30:44 浏览次数:17 分类:博客文章

本文共 1561 字,大约阅读时间需要 5 分钟。

对应着lintcode376题。

 

给定一个二叉树,找出所有路径中各节点相加总和等于给定 目标值 的路径。

一个有效的路径,指的是从根节点到叶节点的路径。

样例

样例1:

输入:

{1,2,4,2,3}
5
输出: [[1, 2, 2],[1, 4]]
说明:
这棵树如下图所示:
1
/ \
2 4
/ \
2 3
对于目标总和为5,很显然1 + 2 + 2 = 1 + 4 = 5
样例2:

输入:

{1,2,4,2,3}
3
输出: []
说明:
这棵树如下图所示:
1
/ \
2 4
/ \
2 3
注意到题目要求我们寻找从根节点到叶子节点的路径。
1 + 2 + 2 = 5, 1 + 2 + 3 = 6, 1 + 4 = 5
这里没有合法的路径满足和等于3.

"""Definition of TreeNode:class TreeNode:    def __init__(self, val):        self.val = val        self.left, self.right = None, None"""class Solution:    """    @param: root: the root of binary tree    @param: target: An integer    @return: all valid paths    """    def binaryTreePathSum(self, root, target):        # write your code here        global res        #保存结果        res=[]        #临时数组        tmp=[]        self.pathSum(res,tmp,root,target)        return res    def pathSum(self,res,tmp,root,target):        if root == None:            return        #首先将当前节点的值加入到tmp中        tmp.append(root.val)        #如果当前节点没有左右孩子,说明是叶子节点        if root.left == None and root.right == None:            #计算根节点到叶子节点的值            s = sum(tmp)            #如果该值等于target            if s == target:                #这里需要拷贝一份tmp,再加入到结果中                newtmp=tmp[:]                res.append(newtmp)        #如果有左孩子        if root.left!=None:            self.pathSum(res,tmp,root.left,target)            #如果遍历到叶子节点了,且不符合sum(tmp)==target,则最后面的节点出栈            tmp.pop()        #如果有右孩子        if root.right!=None:            self.pathSum(res,tmp,root.right,target)            #如果遍历到叶子节点了,且不符合sum(tmp)==target,则最后面的节点出栈            tmp.pop()

 

上一篇:【猫狗数据集】pytorch训练猫狗数据集之创建数据集
下一篇:【python-leetcode111-树的宽度遍历】二叉树的最小深度

发表评论

最新留言

初次前来,多多关照!
[***.217.46.12]2025年04月29日 09时09分18秒