题目描述
给你一棵根为 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]
。
算法分析
dfs
- 算出从根结点到当前点的最大值
maxv
,若当前点val >= maxv
,表示好节点条件成立,res ++
dfs(u,maxv)
:表示从根结点到当前节点的最大值是maxv
,u
表示当前节点
时间复杂度 $O(n)$
Java 代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
static int res = 0;
static void dfs(TreeNode u,int maxv)
{
if(u.val >= maxv) res ++;
//左孩子
if(u.left != null) dfs(u.left,Math.max(maxv,u.left.val));
//右孩子
if(u.right != null) dfs(u.right,Math.max(maxv,u.right.val));
}
public int goodNodes(TreeNode root) {
res = 0;
dfs(root,root.val);
return res;
}
}