GATE CS 2019 简体中文问题19

这是一道涉及树的问题，主要考察树的遍历和节点删除。下面将分几个部分进行介绍。

题干

给定一棵二叉树和一个整数k，你需要从这棵二叉树中删除所有根到叶子路径上节点值之和等于k的路径。要求最后返回删除后的树的根节点。

思路

我们可以使用递归的方式遍历整个树，从根节点开始，对于每个节点，我们分别计算右子树和左子树中从根节点到叶子节点的路径和，如果路径和等于k，则将该节点删除，并返回null节点。

如果该节点是叶子节点，则判断该节点值是否等于k，如果是，则将该节点删除，并返回null节点。

如果该节点不是叶子节点，则对其左右子树分别进行遍历，并将返回的节点赋给该节点的左右子节点。

最后返回根节点即可。

代码

这里给出Java代码实现：

public TreeNode removeNodes(TreeNode root, int k) {
    if (root == null) {
        return null;
    }
    int sum = 0;
    if (root.left == null && root.right == null) {
        sum = root.val;
    }
    TreeNode left = removeNodes(root.left, k - sum);
    TreeNode right = removeNodes(root.right, k - sum);
    if (left == null && right == null && sum == k) {
        return null;
    }
    root.left = left;
    root.right = right;
    return root;
}

复杂度

时间复杂度：O(n)，其中n是二叉树中的节点数。

空间复杂度：O(n)，其中n是二叉树中的节点数。