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

最新推荐文章于 2024-05-20 21:47:59 发布

碎嘴的章鱼哥

最新推荐文章于 2024-05-20 21:47:59 发布

阅读量153

点赞数

文章标签：深度优先 leetcode 算法

本文链接：https://blog.csdn.net/weixin_46765494/article/details/127311890

版权

链接：https://leetcode.cn/problems/count-good-nodes-in-binary-tree/
题目：给你一棵根为 root 的二叉树，请你返回二叉树中好节点的数目。
「好节点」X 定义为：从根到该节点 X 所经过的节点中，没有任何节点的值大于 X 的值。
思路：递归，遍历期间将【好节点】的值作为参数在递归过程中进行传递，并比较当前结点是否大于等于父节点，是则将对应新的【好节点】替换原来的参数值，否则使用原来的【好节点】值继续递归直至结束。题解处返回+1是因为头结点默认是【好节点】，且在遍历过程中没有进行比较。

public static int goodNodes(TreeNode root) {
        int[] result = new int[1];
        dfs(root,root.val,result);
        return result[0]+1;
    }

    public static void dfs(TreeNode root,int max,int[] result){
        if (root==null){
            return;
        }
        if (root.left!=null&&root.left.val>=max){
            result[0]++;
            dfs(root.left,root.left.val,result);
        }else {
            dfs(root.left,max,result);
        }
        if (root.right!=null&&root.right.val>=max){
            result[0]++;
            dfs(root.right,root.right.val,result);
        }else {
            dfs(root.right,max,result);
        }
    }