【二叉树】祖父节点值为偶数的节点和

0x00 题目

给你一棵二叉树
请你返回满足以下条件的所有节点的​​​值之和​​​：
该节点的​​​祖父​​​节点的值为​​偶数​​​ 一个节点的祖父节点是指该节点的​​父节点的父节点​​ 如果不存在祖父节点值为偶数的节点，那么返回 ​​0​​

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 sumEvenGrandparent(_ root: TreeNode?) -> Int {
    guard let root = root else { return 0 }
    
    var res: Int = 0
    func dfs(_ root: TreeNode?, _ p: TreeNode?, _ pp: TreeNode?) {
        guard let r = root else { return }
        
        // 祖父节点存在
        if let node = pp {
            // 是偶数
            if node.val % 2 == 0 {
                res += r.val
            }
        }
        
        dfs(r.left, r, p)
        dfs(r.right, r, p)
    }
    
    dfs(root, nil, nil)
    
    return res
}

0x03 我的小作品

欢迎体验我的作品之一：​​小五笔 86 版​​​ 五笔学习好帮手~
​​App Store​​ 搜索即可~