解决方案通过了 39 个测试用例中的 37 个，但其余测试用例显示“超出时间限制”

Question

我写了以下关于“二叉树的最大深度”的代码。它是一个递归解决方案。我获得了以下必须使用的模板：

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution(object):
    def maxDepth(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """        

鉴于此模板，我的代码如下：

class Solution(object):
    def maxDepth(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """        
        if(not root):
            return 0
        return self.findDepth(root,1)  #self-defined function passing node and depth
        
    def findDepth(self,node,depth):
        
        if(node):
            '''if(node.left and not node.right):
            node=node.left
            if(node.right and not node.left):
            node=node.right'''
            if(node.left or node.right):
                depth=depth+1
            depth=self.findDepth(node.left,depth) if self.findDepth(node.left,depth)>self.findDepth(node.right,depth) else self.findDepth(node.right,depth)  #recursion after computing depth every time
        return depth
        

该解决方案适用于 37 个测试用例，但对于剩余的测试用例，我收到“超出时间限制”的错误。我用谷歌搜索并了解到它的发生是因为代码没有得到适当的优化或有一些不必要的循环/列表/数据结构。我没有使用任何未使用的数据结构，并且代码已尽可能优化。如何摆脱错误消息，以便代码可以在时限内尝试剩余的测试用例？提前致谢。

Answer 1

尽可能优化代码

远非如此，您拨打self.findDepth(...) 的频率是必要的两倍。此外，您强制执行一个您知道将失败的最终递归，只是为了返回depth 参数——if 比方法调用便宜。我期待更像（警告，未经测试的代码）：

class Solution():
    def maxDepth(self, node):
        """
        :type node: TreeNode
        :rtype: int
        """

        depth = 0

        if node:
            left_depth = self.maxDepth(node.left) if node.left else 0
            right_depth = self.maxDepth(node.right) if node.right else 0

            depth = 1 + max(left_depth, right_depth)

        return depth