前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >【python-leetcode113-树的深度遍历】路径总和Ⅱ

【python-leetcode113-树的深度遍历】路径总和Ⅱ

作者头像
西西嘛呦
发布2020-08-26 10:03:01
6420
发布2020-08-26 10:03:01
举报

对应着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.

代码语言:javascript
复制
"""
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()
           
 
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2020-03-02 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档