记录题解:力扣1448. 统计二叉树中好节点的数目

给你一棵根为 root 的二叉树,请你返回二叉树中好节点的数目。

「好节点」X 定义为:从根到该节点 X 所经过的节点中,没有任何节点的值大于 X 的值。

示例 1:

输入:root = [3,1,4,3,null,1,5]
输出:4
解释:图中蓝色节点为好节点。
根节点 (3) 永远是个好节点。
节点 4 -> (3,4) 是路径中的最大值。
节点 5 -> (3,4,5) 是路径中的最大值。
节点 3 -> (3,1,3) 是路径中的最大值。

示例 2:

输入:root = [3,3,null,4,2]
输出:3
解释:节点 2 -> (3, 3, 2) 不是好节点,因为 "3" 比它大。

示例 3:

输入:root = [1]
输出:1
解释:根节点是好节点。

思路:dfs,选出最大值,并计数

# 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 goodNodes(self, root: TreeNode) -> int:
        self.ans = 0

        def dfs(node,maxval):
            
            if node:
                if node.val >= maxval:
                    self.ans += 1
                    maxval = node.val
            if node.left:
                dfs(node.left,maxval)
            if node.right:
                dfs(node.right,maxval)
            # dfs(node.left,maxval)
            # dfs(node.right,maxval)
            return self.ans
        
        return dfs(root, -float('inf'))

遇到的问题:

错误代码:

# 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 goodNodes(self, root: TreeNode) -> int:
        ans = 0

        def dfs(node,maxval):
            
            if node:
                if node.val >= maxval:
                    ans += 1
                    maxval = node.val
            if node.left:
                dfs(node.left,maxval)
            if node.right:
                dfs(node.right,maxval)
            # dfs(node.left,maxval)
            # dfs(node.right,maxval)
            return ans
        
        return dfs(root, -float('inf'))
UnboundLocalError: local variable 'ans' referenced before assignment
    ans += 1
Line 34 in dfs (Solution.py)
    return dfs(root, -float('inf'))
Line 44 in goodNodes (Solution.py)
    ret = Solution().goodNodes(param_1)
Line 63 in _driver (Solution.py)
    _driver()
Line 74 in <module> (Solution.py)

经查找得出以下解决方法:

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

猜你喜欢

转载自blog.csdn.net/weixin_45314061/article/details/130473453