虽然很多人说递归才是符合直觉的写法,但我觉得迭代才是最容易想的。因为这实际上代表了一个控制的视角,代表程序员对代码里的全部细节的掌控。包括有时候写的代码看起来用了递归,但实际上还是迭代的思路,尤其是体现在尾递归上。比如这道题,如果这么写,其实还是迭代的思路:
class Solution:
def __init__(self):
self.depth = 0
def traverse(self, root, depth):
if not root:
self.depth = max(self.depth, depth)
return
self.traverse(root.left, depth + 1)
self.traverse(root.right, depth + 1)
def maxDepth(self, root: Optional[TreeNode]) -> int:
self.traverse(root, 0)
return self.depth
因为从本质上来说,还是迭代那种维护变量并更新的思路,这意味着程序员希望对全局有一个掌控。不过,其实完全可以把过程视作黑盒,只考虑输入输出,把整个过程当成函数的重复和组合:
class Solution:
def maxDepth(self, root: Optional[TreeNode]) -> int:
if not root:
return 0
return max(self.maxDepth(root.left), self.maxDepth(root.right)) + 1
这并不意味着解法有优劣之分(虽然性能上可能有区别,但一般不会有本质区别),代表的只是一种思维方式的不同。所以我用了“控制反转”这个词,意味着程序员是否愿意将自己对代码的一部分控制权交出去;这可能是比题目本身值得思考的问题。