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
搜索即可~