链接: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);
}
}
作者小白,侵权请私信删
2022.10.13