Question

请检查我的代码。我找不到bug的位置。问题是here。

这是我的解决方案：

# Given a binary tree, find all paths that sum of the nodes in the path equals to a given number target.
#
# A valid path is from root node to any of the leaf nodes.

# Example
# Given a binary tree, and target = 5:
#
#      1
#     / \
#    2   4
#   / \
#  2   3
# return
#
# [
#   [1, 2, 2],
#   [1, 4]
# ]



class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

class Solution:
    # @param {TreeNode} root the root of binary tree
    # @param {int} target an integer
    # @return {int[][]} all valid paths

    result = []

    def binaryTreePathSum(self, root, target):
        # Write your code here
        if root is None:
            return self.result
        self.dfs(root, [], target)
        return self.result

    def dfs(self, node, tmp, target):
        tmp.append(node.val)
        if node.left is None and node.right is None:
            if target == sum(tmp):
                #print tmp
                self.result.append(tmp)
            tmp.pop()
            return

        if node.left:
            self.dfs(node.left, tmp, target)
        if node.right:
            self.dfs(node.right, tmp, target)
        tmp.pop()

if __name__ == "__main__":
    root = TreeNode(1)
    root.left = TreeNode(2)
    root.left.left = TreeNode(2)
    root.left.right = TreeNode(3)
    root.right = TreeNode(4)
    result = Solution().binaryTreePathSum(root, 5)
    print result

假设输入为{1,2,4,2,3}, 5。运行解决方案后，输出为[[],[]]。但如果我取消print tmp，则输出将为

[1, 2, 2]
[1, 4]
[[],[]]

tmp的输出是正确的。但似乎result.append(tmp)未将tmp附加到result。我不知道问题出在哪里。

Answer 1

问题是，您的result列表多次包含一个且相同的列表。

您将tmp列表附加到result，如下所示：

self.result.append(tmp)

然后，您使用tmp.pop()和tmp.append(...)改变了同一个列表。这就是为什么最后，你的所有结果都是空列表。

解决方案很简单，只需创建列表的副本：

self.result.append(tmp[:])

二叉树路径求和（python）

1 个答案: