原题
给你一棵根为 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 解释:根节点是好节点。
提示:
- 二叉树中节点数目范围是
[1, 10^5]
。 - 每个节点权值的范围是
[-10^4, 10^4]
。
解题思路
显然我们需要遍历每个节点,并且将当前路径上最大值一起往下传,是一个和路径深度有关系的问题,考虑使用DFS实现遍历:文章来源:https://www.toymoban.com/news/detail-673246.html
class Solution {
public:
int goodNodes(TreeNode* root) {
return dfs(root);
}
void dfs(TreeNode* root) {
if(root == nullptr){
return;
}
visit(root);
dfs(root->left);
dfs(root->right);
}
};
假设我们当前有一个节点root和路径上传下来的最大值path_max。如果当前节点值小于path_max,那我们继续把path_max传给它的左右节点。如果当前节点值大于path_max,则把path_max设为当前节点值,并且把path_max传给它的左右节点。而当前节点的好节点数,等于左右节点的号节点数之和,加上1,如果当前节点是好节点,否则不加。考虑到根节点一定是好节点,我们可以用INT_MIN作为path_max的初始值。于是我们得到完整的实现:文章来源地址https://www.toymoban.com/news/detail-673246.html
class Solution {
public:
int goodNodes(TreeNode* root) {
return dfs(root,INT_MIN);
}
void dfs(TreeNode* root, int path_max) {
if(root == nullptr){
return 0;
}
int res = 0;
if(root->val >= path_max){
res++;
path_max = root->val;
}
res += dfs(root->left, path_max);
res += dfs(root->right, path_max);
return res;
}
};
到了这里,关于Leetcode每日一题:1448. 统计二叉树中好节点的数目的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!