@author: sdubrz
@date: 2020.04.15
题目难度: 中等
考察内容: 树
原题链接 https://leetcode-cn.com/problems/sum-of-nodes-with-even-valued-grandparent/
题目的著作权归领扣网络所有,商业转载请联系官方授权,非商业转载请注明出处。
解题代码转载请联系 lwyz521604#163.com
给你一棵二叉树,请你返回满足以下条件的所有节点的值之和:
- 该节点的祖父节点的值为偶数。(一个节点的祖父节点是指该节点的父节点的父节点。)
如果不存在祖父节点值为偶数的节点,那么返回 0 。
示例:
输入:root = [6,7,8,2,7,1,3,9,null,1,4,null,null,null,5]
输出:18
解释:图中红色节点的祖父节点的值为偶数,蓝色节点为这些红色节点的祖父节点。
提示:
- 树中节点的数目在 1 到 10^4 之间。
- 每个节点的值在 1 到 100 之间。
通过次数3,320提交次数4,138
解法
最简单的方式用一个递归地方式即可:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public int sumEvenGrandparent(TreeNode root) {
if(root==null){
return 0;
}
int sum = 0;
if(root.val%2==0){
sum += this.sumEvenParent(root.left);
sum += this.sumEvenParent(root.right);
}
sum += this.sumEvenGrandparent(root.left);
sum += this.sumEvenGrandparent(root.right);
return sum;
}
// 返回node的两个孩子的数值之和
private int sumEvenParent(TreeNode node){
if(node==null){
return 0;
}
int sum = 0;
if(node.left!=null){
sum += node.left.val;
}
if(node.right!=null){
sum += node.right.val;
}
return sum;
}
}
下面是在 LeetCode 系统中的提交结果。因为 sumEvenParent
函数导致了部分重复的遍历,所以时间上并不是最有效的方法。如果要加速可以 sumEvenParent
的功能合并到 sumEvenGrandparent
中去,不过可能会造成空间复杂度的稍微提升。
执行结果: 通过 显示详情
执行用时 : 2 ms, 在所有 Java 提交中击败了 68.56% 的用户
内存消耗 : 41.5 MB, 在所有 Java 提交中击败了 100.00% 的用户