113. 路径总和 II [medium]

https://leetcode-cn.com/problems/path-sum-ii/

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

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

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

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1

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

通过次数59,603 | 提交次数98,721

First Try

2020-07-17

对backtrack回溯递归比较熟悉后，很快就能搞定。

重要一是判断叶子节点，并在保存临时路径的时候用上复制；二是每次递归后都需要对path进行复原。

class Solution(object):
    def pathSum(self, root, sum):
        """
        :type root: TreeNode
        :type sum: int
        :rtype: List[List[int]]
        """

        def dfs(root, score, target, path, rv):
            if not root:
                return 
            score += root.val 
            if (not (root.left or root.right)) and score == target:
                rv.append(path[:])  # 需要复制一波
            if root.left:
                path.append(root.left.val)
                dfs(root.left, score, target, path, rv)
                path.pop()
            if root.right:
                path.append(root.right.val)
                dfs(root.right, score, target, path, rv)
                path.pop()

        if not root:
            return []

        rv = []
        dfs(root, 0, sum, [root.val], rv)
        return rv 

执行用时：24 ms, 在所有 Python 提交中击败了97.66%的用户
内存消耗：15.7 MB, 在所有 Python 提交中击败了50.00%的用户

113. 路径总和 II [medium]​

First Try​

113. 路径总和 II [medium]

First Try