祖父节点值为偶数的节点和

1315. 祖父节点值为偶数的节点和

难度中等49收藏分享切换为英文接收动态反馈

给你一棵二叉树，请你返回满足以下条件的所有节点的值之和：

• 该节点的祖父节点的值为偶数。（一个节点的祖父节点是指该节点的父节点的父节点。）

如果不存在祖父节点值为偶数的节点，那么返回 0 。

示例：

输入：root = [6,7,8,2,7,1,3,9,null,1,4,null,null,null,5]
输出：18
解释：图中红色节点的祖父节点的值为偶数，蓝色节点为这些红色节点的祖父节点。

提示：

• 树中节点的数目在 1 到 10^4 之间。
• 每个节点的值在 1 到 100 之间。

深度优先遍历

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    int sum = 0;
    public int sumEvenGrandparent(TreeNode root) {
        deep(root, 0, new HashSet<>());
        return sum;
    }

    public void deep(TreeNode root, int level, Set<Integer> set) {
        if (root == null) return;
        if (set.contains(level)) {
            sum += root.val;
        }
        Integer l = null;
        if (root.val % 2 == 0) {
            l = level + 2;
            set.add(l);
        }
        
        deep(root.left, level + 1, set);
        deep(root.right, level + 1, set);
        if (l != null)set.remove(l);
    }
}

赏

谢谢你请我吃糖果

微信