【发布时间】:2019-08-23 04:58:56
【问题描述】:
我正在尝试解决 leetcode 问题 #113,即“给定一个二叉树和一个总和,找到所有从根到叶的路径,其中每个路径的总和等于给定总和”
我的问题是为什么下面显示的代码 #1 会打印树中所有节点的值?递归堆栈在代码 #1 中的工作方式与递归堆栈在代码 #2 中的工作方式相反,这是正确的解决方案?
非常感谢您对我的帮助!
#code #1
class Solution:
def pathSum (self, root, sum):
self.res = []
self.dfs(root, sum, [])
return self.res
def dfs(self, root, sum, path):
if not root:
return
sum -= root.val
path += [root.val]
if not root.left and not root.right and sum == 0:
self.res.append(path)
self.dfs(root.left, sum, path)
self.dfs(root.right, sum, path)
#code #2
class Solution:
def pathSum (self, root, sum):
self.res = []
self.dfs (root, sum, [])
return self.res
def dfs (self, root, sum, path):
if not root:
return
sum -= root.val
if not root.left and not root.right and sum == 0:
self.res.append(path + [root.val])
self.dfs(root.left, sum, path+[root.val])
self.dfs(root.right, sum, path+[root.val])
【问题讨论】:
-
这个问题似乎是开放式的,而且相当不精确。写下一些例子并在上面测试你的代码。它应该让你对你的代码到底在做什么有一些看法。
-
@fulaphex 您好,感谢您对这个问题的关注!我很乐意写下一些例子并在上面测试我的代码。但问题是我不确定递归堆栈是如何与这段代码一起工作的,所以我无法测试它。
-
我可以建议两种方法: 1. 在一张纸上写下一些示例树并分析纸上的代码。 2. 为节点实现缺失的类,写下示例树并使用调试器或一些调试打印语句分析程序。
标签: python python-3.x python-2.7 recursion