跟着专注于计算机视觉的AndyJ的妈妈我学算法之每日一题leetcode113路径总和 II

题目:

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

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

示例:
给定如下二叉树,以及目标和 sum = 22,

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1
返回:

[
   [5,4,11,2],
   [5,8,4,5]
]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/path-sum-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

回溯,写就是了。

class Solution:
    def pathSum(self, root: TreeNode, target: int) -> List[List[int]]:
        self.ret = []
        def dfs(root, res):
            if not root:
                return
            if not root.left and not root.right:
                if (sum(res) + root.val) == target:
                    self.ret.append(res[:]+[root.val])
                return
            
            res.append(root.val)
            dfs(root.left, res)
            dfs(root.right, res)
            res.pop()
        dfs(root, [])
        return self.ret

好了。

猜你喜欢

转载自blog.csdn.net/mianjiong2855/article/details/107314412