【Leetcode】1315. Sum of Nodes with Even-Valued Grandparent

题目地址：

https://leetcode.com/problems/sum-of-nodes-with-even-valued-grandparent/

给定一棵二叉树，求所有祖父节点为偶数的节点数值之和。

思路是DFS。只需要把节点的深度、父亲和祖父节点作为参数，在DFS的时候进行传递即可。代码如下：

public class Solution {
    public int sumEvenGrandparent(TreeNode root) {
        return dfs(root, 0, root, root);
    }
    
    private int dfs(TreeNode cur, int depth, TreeNode parent, TreeNode grandParent) {
        if (cur == null) {
            return 0;
        }
        
        int sum = 0;
        // 如果深度大于等于2，并且祖父是偶数，就将本节点数值加入sum中
        if (depth >= 2 && grandParent.val % 2 == 0) {
            sum += cur.val;
        }
        
        // 分治，分别收集左右子树上满足条件的节点和
        // 进入下一层遍历时，当前节点变成了父亲节点，父亲节点变成了祖父节点
        sum += dfs(cur.left, depth + 1, cur, parent);
        sum += dfs(cur.right, depth + 1, cur, parent);
        
        return sum;
    }
}

class TreeNode {
    int val;
    TreeNode left, right;
    TreeNode(int x) {
        val = x;
    }
}

时间复杂度$O(n)$，空间$O(h)$。

C++：

class Solution {
 public:
  int sumEvenGrandparent(TreeNode* root) { return dfs(root, 0, root, root); }

  int dfs(TreeNode* cur, int depth, TreeNode* p, TreeNode* gp) {
    if (!cur) return 0;
    int sum = 0;
    if (depth >= 2 && gp->val % 2 == 0) sum += cur->val;
    return sum + dfs(cur->left, depth + 1, cur, p) +
           dfs(cur->right, depth + 1, cur, p);
  }
};

时空复杂度一样。