二叉树-路径总和

题目来源:https://leetcode-cn.com/problems/path-sum-ii/

注:112与本题基本一样,稍微改一下就可以了 https://leetcode-cn.com/problems/path-sum/

给定一个二叉树和一个目标和,找到所有从根节点到叶子节点路径总和等于给定目标和的路径。

说明: 叶子节点是指没有子节点的节点。

首先,从题目的“从根节点到叶子节点”这一句,可以很自然的想到深度先序遍历,按根-左子数-右子树的顺序访问。

那么就可以采用先序遍历的形式,每次将当前节点的值记录下来,传递给当前节点的左右子树,直到叶节点,则判断哪一条路径的节点列表和为target。

解法一:非递归(比较好理解)

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def pathSum(self, root, target):
        """
        :type root: TreeNode
        :type target: int
        :rtype: List[List[int]]
        """
        if not root:
            return []
        stack = [([root.val], root)]     # 先序遍历,根先入栈,注意存值用list,方便后面直接对list进行sum
        ans = []
        while stack:
            val, tmp = stack.pop() 
            if tmp.right: 
                stack.append((val + [tmp.right.val], tmp.right))
            if tmp.left:
                stack.append((val + [tmp.left.val], tmp.left))
            if not tmp.right and not tmp.left and sum(val) == target:    # 叶节点,且路径节点值的和为目标
                ans.append(val)         # 增加路径节点值
        return ans

解法二:递归

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def pathSum(self, root, target):
        """
        :type root: TreeNode
        :type target: int
        :rtype: List[List[int]]
        """
        ans = []
        def dfs(root, target, tmp):
            if not root:
                return 
            if not root.right and not root.left and target-root.val == 0:  # 叶节点,且target与路径节点值相减后为0
                tmp += [root.val]           # 路径节点值加上当前节点(叶节点)
                ans.append(tmp)             # 满足条件的路径
            dfs(root.left, target - root.val, tmp + [root.val])   # 每次递归时,target减去已经访问过的节点的值,tmp加上这个值
            dfs(root.right, target - root.val, tmp + [root.val])
        dfs(root, target, [])
        return ans

 

参考:

https://leetcode-cn.com/problems/path-sum-ii/solution/dfsfei-di-gui-python3-by-baiyizhe/

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值