UnboundLocalError: local variable ‘ans‘ referenced before assignment

最新推荐文章于 2024-10-16 16:28:57 发布

叫我Jesslyn就好了

最新推荐文章于 2024-10-16 16:28:57 发布

阅读量927

点赞数 1

分类专栏： python报错文章标签： leetcode

本文链接：https://blog.csdn.net/m0_50233720/article/details/115917982

版权

python报错专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一种求解二叉树中最长相同值路径的算法，并对其进行了修正。通过递归遍历左子树和右子树，计算路径长度并更新全局变量以找到最长路径。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

def longestUnivaluePath(self, root: TreeNode) -> int:
        if not root:
            return 0
        ans = 0
        def longr(root):
            if not root:
                return 0
            pl = 0
            pr = 0
            
            l = longr(root.left)
            r = longr(root.right)
            
            if root.left and root.val == root.left.val:
                pl = l + 1
            if root.right and root.val == root.right.val:
                pr = r + 1

            ans = max(ans,pl + pr)
            return max(pl,pr)
        longr(root)
        return ans

问题出在ans没有被reference
需要把function外的实参ans传入function内，应该在实参加上self.ans,形参也得加上self.ans

修改后

 def longestUnivaluePath(self, root: TreeNode) -> int:
        if not root:
            return 0
        self.ans = 0
        def longr(root):
            if not root:
                return 0
            pl = 0
            pr = 0
            
            l = longr(root.left)
            r = longr(root.right)
            
            if root.left and root.val == root.left.val:
                pl = l + 1
            if root.right and root.val == root.right.val:
                pr = r + 1

            self.ans = max(self.ans,pl + pr)
            return max(pl,pr)
        longr(root)
        return self.ans