缺少的4是因为您从未附加它。在你的成功案例中:if root.val == k:
return l
…你需要这样做:
^{pr2}$
{{2}在这种情况下,的值总是附加到中。在
只有在递归调用中的任何一个返回非空列表时,才可以通过追加它来解决这个问题,但这样当然会使节点失去顺序。最简单的解决方法是故意让节点无序:# Pre-order traversal: Visit root, then left, then right.
if path(root.left, k, l) or path(root.right, k, l):
l.append(root.val)
…然后在末尾反转列表,例如,在包装器函数中:def path2(root, k):
return list(reversed(path(root, k)))
但是,代码中还有一个问题,就在这里:def path(root, k, l=[]):
当[]执行def时,[]即l的默认值,并在每次调用中重用。这意味着path2(a, 4)将第一次返回正确的答案,[1, 2, 4],但当您第二次调用它时,它将继续追加到同一个列表并返回[1, 2, 4, 1, 2, 4]。在
有两种惯用的方法来解决这个问题,但是在我们的例子中,因为我们已经在使用path2包装函数,所以我们最好在这里修复它:def path2(root, k):
return list(reversed(path(root, k, [])))
…然后删除path上的默认值。在
但是,您可能需要考虑重新开始使用更易于理解的版本:def path(root, k):
if not root:
return []
if root.val == k:
return [root.val]
res = path(root.left, k)
if res:
return [root.val] + res
res = path(root.right, k)
if res:
return [root.val] + res
return []
(你可以把它缩短一点;我特意在这里把一切都说得很清楚。)
对于许多递归问题,反转它们并传递累加器是一个重要的优化,因为尾部调用消除可以从堆栈中删除其中一个分支。尽管Python不做TCE,但是仍然值得学习如何用tail-calling的方式来做事情,这只是为了你自己的理解(当然,如果你用另一种语言编写代码的话)。但是首先要学会如何做简单的版本,然后再尝试反转它。在