笔试问题4

关于局部变量及全局变量的问题：

力扣543：二叉树的最大直径

给定一棵二叉树，你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。

示例 :
给定二叉树

          1
         / \
        2   3
       / \     
      4   5    
返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。

注意：两结点之间的路径长度是以它们之间边的数目表示。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/diameter-of-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

代码如下：

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def diameterOfBinaryTree(self, root: Optional[TreeNode]) -> int:
        maxdep = 0
        def maxdepth(root):
            global maxdep
            if root is None:
                return 0
            leftmax = maxdepth(root.left)
            rightmax = maxdepth(root.right)
            curdep = leftmax + rightmax
            maxdep = max(curdep,maxdep)
            return 1 + max(leftmax,rightmax)
        maxdepth(root)
        return maxdep

 刚开始，会进行报错，报错如下：

UnboundLocalError: local variable 'maxdep' referenced before assignment

问题还是在于变量在函数中修改时引发的问题，修改方法如下：

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def diameterOfBinaryTree(self, root: Optional[TreeNode]) -> int:
        self.maxdep = 0
        def maxdepth(root):
            if root is None:
                return 0
            leftmax = maxdepth(root.left)
            rightmax = maxdepth(root.right)
            curdep = leftmax + rightmax
            self.maxdep = max(curdep,self.maxdep)
            return 1 + max(leftmax,rightmax)
        maxdepth(root)
        return self.maxdep