Leetcode每日一题: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
解释:根节点是好节点。

提示:

  • 二叉树中节点数目范围是 [1, 10^5] 。
  • 每个节点权值的范围是 [-10^4, 10^4] 。

解题思路

显然我们需要遍历每个节点,并且将当前路径上最大值一起往下传,是一个和路径深度有关系的问题,考虑使用DFS实现遍历:

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的初始值。于是我们得到完整的实现:

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;
    }
};

猜你喜欢

转载自blog.csdn.net/Mamong/article/details/132496104
今日推荐