给你一棵根为 root
的二叉树,请你返回二叉树中好节点的数目。
「好节点」X 定义为:从根到该节点 X 所经过的节点中,没有任何节点的值大于 X 的值。
输入: root = [3,1,4,3,null,1,5] 输出: 4 解释: 图中蓝色节点为好节点。 根节点 (3) 永远是个好节点。 节点 4 -> (3,4) 是路径中的最大值。 节点 5 -> (3,4,5) 是路径中的最大值。 节点 3 -> (3,1,3) 是路径中的最大值。
输入: root = [3,3,null,4,2] 输出: 3 解释: 节点 2 -> (3, 3, 2) 不是好节点,因为 "3" 比它大。
输入: root = [1] 输出: 1 解释: 根节点是好节点。
- 二叉树中节点数目范围是
[1, 10^5]
。 - 每个节点权值的范围是
[-10^4, 10^4]
。
# 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:
def foo(root: TreeNode, max_val: int) -> int:
max_val = max(max_val, root.val)
ret = 1 if root.val >= max_val else 0
ret += foo(root.left, max_val) if root.left else 0
ret += foo(root.right, max_val) if root.right else 0
return ret
return foo(root, -10000)