为什么我的二叉树的最大深度解决方案返回的比预期的少一？答案

【问题标题】：Why is my maximum depth solution for binary tree returning one less than it is supposed to?为什么我的二叉树的最大深度解决方案返回的比预期的少一？
【发布时间】：2021-07-26 03:52:06
【问题描述】：

我正在解决以下 Leetcode 问题： https://leetcode.com/problems/maximum-depth-of-binary-tree/solution/

就是返回二叉树的最大深度。

这是我的解决方案：

class Solution:
    def maxDepth(self, root: TreeNode) -> int:
        
        if not root:
            return 0

        stack = [(root, 1)]
        
        curr_depth = 1
        while stack:
            node, depth = stack.pop()
            depth = max(curr_depth,depth)
            if node.right:
                stack.append((node.right, curr_depth + 1))
            if node.left:
                stack.append((node.left, curr_depth + 1))
        
        return depth

由于某种原因，输出总是比预期的少一。并且查看接受的解决方案，它们看起来与我的非常相似，但我似乎无法找到我的解决方案出错的地方。

【问题讨论】：

您在使用调试器时发现了什么？
node, depth = stack.pop() 应该是 node, curr_depth 吗？
哦，好吧，是的，我现在看到了 jamesdlin。
@mkrieger1 - 调试器没有显示错误。我在 Pycharm 上运行它
感谢 mkrieger1 的链接，我会仔细阅读它们——非常有用！

标签： python binary-tree

【解决方案1】：

这有帮助吗？

from typing import Optional

class Solution:
    def maxDepth(self, root: TreeNode) -> int:
        return self._visit(root, 0)
    
    def _visit(self, node: Optional[TreeNode], depth: int) -> int:
        if not node:
            return depth
        return max(self._visit(node.left, depth + 1), self._visit(node.right, depth + 1))

【讨论】：

【解决方案2】：

所以你的违规变量是深度和curr_depth。基本上当depth = max(curr_depth,depth) 发生时，您希望将其分配给curr_depth。然后在添加到堆栈时您要传递depth+1 而不是curr_depth+1，最后您要返回curr_depth 而不是depth。 curr_depth 表示当前最大深度，而depth 是一个临时变量，表示所考虑节点的深度。最好保持一致。

我不会浪费您的时间并复制和粘贴正确的有效算法，正如您所说，您的问题是关于您的算法失败的原因，因为您已经看到了许多有效的实现。

【讨论】：