yankewei / LeetCode

LeetCode 问题的解决方法
MIT License
6 stars 0 forks source link

统计二叉树中好节点的数目 #165

Open yankewei opened 1 year ago

yankewei commented 1 year ago

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

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

示例 1:

示例1

输入:root = [3,1,4,3,null,1,5]
输出:4
解释:图中蓝色节点为好节点。
根节点 (3) 永远是个好节点。
节点 4 -> (3,4) 是路径中的最大值。
节点 5 -> (3,4,5) 是路径中的最大值。
节点 3 -> (3,1,3) 是路径中的最大值。

示例 2:

示例2

输入:root = [3,3,null,4,2]
输出:3
解释:节点 2 -> (3, 3, 2) 不是好节点,因为 "3" 比它大。

示例 3:

示例3

输入:root = [1]
输出:1
解释:根节点是好节点。

提示:

yankewei commented 1 year ago

很简单的深度优先搜索,每次遍历到一个节点,如果该节点的值大于等于这条路线的最大值,那就就返回

/**
 * Definition for a binary tree node.
 * class TreeNode {
 *     public $val = null;
 *     public $left = null;
 *     public $right = null;
 *     function __construct($val = 0, $left = null, $right = null) {
 *         $this->val = $val;
 *         $this->left = $left;
 *         $this->right = $right;
 *     }
 * }
 */
class Solution {

    /**
     * @param TreeNode $root
     * @return Integer
     */
    function goodNodes($root) {
        return $this->dfs(PHP_INT_MIN, $root);
    }

    function dfs($max_val, $root) {
        if ($root === null) {
            return 0;
        }

        $ret = 0;

        if ($root->val >= $max_val) {
            $ret += 1;
            $max_val = $root->val;
        }

        $ret += $this->dfs($max_val, $root->left) + $this->dfs($max_val, $root->right);
        return $ret;
    }
}