Question

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

sum = 25
  ans = []
  def recurse(s, traverse, ls):
                #append tuple
                ls = ls + (traverse.val,)
                if (traverse.right):
                    recurse(s, traverse.right, ls)
                if (traverse.left):
                    recurse(s, traverse.left, ls)
                if (s==sum): 
                    #convert touple and add to answer
                    tmp = list(ls)
                    ans.append(tmp)

  a = tuple()
  recurse(0, root, a)

在这段代码中，我动态地将列表附加到列表（ans）。在这个例子中，我使用递归并使用元组来构建每个子列表，因为它是不可变的。然后，我将其转换为列表并将其附加到我的答案。如果我只使用列表，那么每次调用函数时都无法传递副本。

以这种方式使用元组是否有任何性能下降？有没有更好的方法来递归创建类似问题的列表？

此外，这种方法不适用于2d列表...什么是制作2D不可变列表的最佳方法？

Answer 1

我猜sum是您想要进行的最大遍历次数。

以下将解决这个问题：

def traverse(root, limit=None, count=0):
    if not root or count == limit:
        return []
    traversal = [root.val]
    traversal += traverse(root.right, limit, count + len(traversal))
    traversal += traverse(root.left, limit, count + len(traversal))
    return traversal

然后，你只需要这样称呼它：

root = TreeNode(20)
root.right = TreeNode(40)
root.right.right = TreeNode(50)
root.right.left = TreeNode(30)
root.left = TreeNode(10)
root.left.left = TreeNode(5)

traverse(root)          // [20, 40, 50, 30, 10, 5]
traverse(root, limit=1) // [20]
traverse(root, limit=4) // [20, 40, 50, 30]

或者，如果您可以在right提供left和__init__个节点，那么您可以采用一种很酷的功能方式：

print(
    traverse(
        Node(20,
            right=Node(40,
                right=Node(50),
                left=Node(30)),
            left=Node(10,
                left=Node(5))),
        limit=3))
 # > [20, 40, 50]

Python - 递归创建列表的最佳方法？

1 个答案: