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

原创

haocold 2022-10-24 19:20:44 博主文章分类：算法 ©著作权

文章标签 swift 二叉树算法当前路径搜索 文章分类 运维

©著作权归作者所有：来自51CTO博客作者haocold的原创作品，请联系作者获取转载授权，否则将追究法律责任

0x00 题目

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

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

0x01 思路

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

0x02 解法

语言：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
}