【Leetcode】1448. Count Good Nodes in Binary Tree

记录算法题解

已于 2022-08-29 11:56:19 修改

阅读量292

点赞数

分类专栏： LC 树、分治与堆文章标签：二叉树 leetcode 算法 dfs

于 2020-11-16 05:34:16 首次发布

本文链接：https://blog.csdn.net/qq_46105170/article/details/109713258

版权

LC 树、分治与堆专栏收录该内容

296 篇文章 4 订阅

订阅专栏

该博客介绍了如何解决LeetCode上的一个问题，即计算二叉树中好节点的数量。好节点是指从根到该节点路径上的所有节点中值最大的那个节点。解决方案使用了深度优先搜索（DFS），在遍历过程中维护路径上的最大值，并在当前节点值大于等于最大值时增加计数。代码分别用Java和C++实现，时间复杂度为O(n)，空间复杂度为O(h)。

摘要由CSDN通过智能技术生成

题目地址：

https://leetcode.com/problems/count-good-nodes-in-binary-tree/

给定一棵二叉树，如果一个节点是从树根到其路径的所有节点的最大值，则称其为“好节点”。求这棵二叉树的好节点个数。

思路是DFS。在DFS的时候将到达当前节点的路径的所有节点的最大值传递到下一层递归里去。代码如下：

public class Solution {
    
    private int res;
    
    public int goodNodes(TreeNode root) {
        if (root == null) {
            return 0;
        }
        
        res = 0;
        dfs(root, Integer.MIN_VALUE);
        return res;
    }
    
    private void dfs(TreeNode cur, int max) {
        if (cur == null) {
            return;
        }
        
        // 如果当前节点是从根到当前节点路径所有节点的最大值，则计入计数
        if (cur.val >= max) {
            res++;
        }
        
        max = Math.max(cur.val, max);
        dfs(cur.left, max);
        dfs(cur.right, max);
    }
}

class TreeNode {
    int val;
    TreeNode left, right;
    
    public TreeNode(int val) {
        this.val = val;
    }
}

时间复杂度 $O (n)$ ，空间 $O (h)$ 。

C++：

class Solution {
 public:
  int goodNodes(TreeNode* root) {
    int res = 0;
    dfs(root, INT_MIN, res);
    return res;
  }

  void dfs(TreeNode* cur, int maxv, int &res) {
    if (!cur) return;
    if (cur->val >= maxv) res++;
    int x = max(maxv, cur->val);
    dfs(cur->left, x, res);
    dfs(cur->right, x, res);
  }
};