【二叉树】统计二叉树中好节点的数目

本文介绍: 题目给你一棵根为 root 的二叉树请你返回二叉树中好节点的数目「好节点」X 定义为：从根到该节点 X 所经过的节点中没有任何节点的值大于 X 的值

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

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

经过一个节点时
计算当前路径的最大值
并传递给下一个节点
通过比较即可判断当前节点是否符合要求

语言：Swift

树节点：TreeNode

public class TreeNode {
    public var val: Int
    public var left: TreeNode?
    public var right: TreeNode?
    public init() { self.val = 0; self.left = nil; self.right = nil; }
    public init(_ val: Int) { self.val = val; self.left = nil; self.right = nil; }
    public init(_ val: Int, _ left: TreeNode?, _ right: TreeNode?) {
        self.val = val
        self.left = left
        self.right = right
    }
}

解法：

func goodNodes(_ root: TreeNode?) -> Int {
    guard let root = root else { return 0 }
    var res: Int = 0
    
    func dfs(_ root: TreeNode?, _ val: Int) {
        guard let root = root else { return }

        // 是否是好节点
        if root.val >= val { res += 1}
        
        // 计算该条路径的最大值
        let big = max(root.val, val)
        
        // 递归左右节点
        dfs(root.left, big)
        dfs(root.right, big)
    }

    dfs(root, Int.min)
    
    return res
}