Leetcode 1448. 统计二叉树中好节点的数目

原创于 2021-02-13 13:32:41 发布 · 159 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#深度优先遍历 #C++

LeetCode 专栏收录该内容

210 篇文章

订阅专栏

给定一棵二叉树，好节点是指从根节点到该节点路径上没有比它值大的节点。该算法实现通过深度优先搜索（DFS）遍历二叉树，计算每个节点是否为好节点，并返回好节点的总数。

题目描述

给你一棵根为 root 的二叉树，请你返回二叉树中好节点的数目。

「好节点」X 定义为：从根到该节点 X 所经过的节点中，没有任何节点的值大于 X 的值。

C++

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int goodNodes(TreeNode* root) {
        if(root==NULL){
            return 0;
        }
        return 1+dfs(root->left,root->val)+dfs(root->right,root->val); 
    }
    int dfs(TreeNode* root,int parent){
        if(root==NULL){
            return 0;
        }
        if(root->val>=parent){
            return 1+dfs(root->left,root->val)+dfs(root->right,root->val);
        }else{
            return   dfs(root->left,parent)+dfs(root->right,parent);
        }    
    }
};