给你一棵根为 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