给你一棵根为 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) 是路径中的最大值。
1. 求根节点到当前结点的最大值,采用前序遍历完成。
2. 从根节点到当前结点的最大值需要能够进行相应回溯,因此需要将最大值设为递归函数的参数传递下去
class Solution {
public:
int goodNodes(TreeNode* root) {
return dfs(-10010, root);
}
int dfs(int maxx, TreeNode* root){
if (root == nullptr) return 0;
maxx = max(maxx, root->val);
if (maxx == root->val) return 1 + dfs(maxx, root->left) + dfs(maxx, root->right);
else return dfs(maxx, root->left) + dfs(maxx, root->right);
}
};