LC 985. 查询后的偶数和

题目描述

这是 LeetCode 上的 985. 查询后的偶数和 ，难度为 中等。

给出一个整数数组 A 和一个查询数组 queries。

对于第 i 次查询，有 val = queries[i][0], index = queries[i][1]，我们会把 val 加到 A[index] 上。然后，第 i 次查询的答案是 A 中偶数值的和。

此处给定的 index = queries[i][1] 是从 0 开始的索引，每次查询都会永久修改数组 A。

返回所有查询的答案，你的答案应当以数组 answer 给出，answer[i] 为第 i 次查询的答案。

示例：

输入：A = [1,2,3,4], queries = [[1,0],[-3,1],[-4,0],[2,3]]

输出：[8,6,2,4]

解释：
开始时，数组为 [1,2,3,4]。
将 1 加到 A[0] 上之后，数组为 [2,2,3,4]，偶数值之和为 2 + 2 + 4 = 8。
将 -3 加到 A[1] 上之后，数组为 [2,-1,3,4]，偶数值之和为 2 + 4 = 6。
将 -4 加到 A[0] 上之后，数组为 [-2,-1,3,4]，偶数值之和为 -2 + 4 = 2。
将 2 加到 A[3] 上之后，数组为 [-2,-1,3,6]，偶数值之和为 -2 + 6 = 4。

提示：

• $1 <= A.length <= 10000$
• $-10000 <= A[i] <= 10000$
• $1 <= queries.length <= 10000$
• $-10000 <= queries[i][0] <= 10000$
• $0 <= queries[i][1] < A.length$

---

模拟

为了方便，将 queries 记为 qs。

由于每次修改都是对 A 的永久修改，因此我们使用一个变量 t 来记录当前所有偶数和即可。

具体的，我们先统计原数组所有偶数和为 t，从前往后处理每个 $qs[i]$，若修改前的原值 $nums[qs[i][1]]$ 为偶数，我们需要将其从 t 中减去，再进行实际修改 nums[qs[i][1]] += qs[i][0]，若修改后值为偶数，则将其累加到 t 上。

Java 代码：

class Solution {
    public int[] sumEvenAfterQueries(int[] nums, int[][] qs) {
        int m = qs.length, t = 0;
        int[] ans = new int[m];
        for (int x : nums) t += x % 2 == 0 ? x : 0;
        for (int i = 0; i < m; i++) {
            int val = qs[i][0], idx = qs[i][1];
            if (nums[idx] % 2 == 0) t -= nums[idx];
            nums[idx] += val;
            if (nums[idx] % 2 == 0) t += nums[idx];
            ans[i] = t;
        }
        return ans;
    }
}

C++ 代码：

class Solution {
public:
    vector<int> sumEvenAfterQueries(vector<int>& nums, vector<vector<int>>& qs) {
        int m = qs.size(), t = 0;
        std::vector<int> ans(m);
        for (int x : nums) t += x % 2 == 0 ? x : 0;
        for (int i = 0; i < m; i++) {
            int val = qs[i][0], idx = qs[i][1];
            if (nums[idx] % 2 == 0) t -= nums[idx];
            nums[idx] += val;
            if (nums[idx] % 2 == 0) t += nums[idx];
            ans[i] = t;
        }
        return ans;
    }
};

Python3 代码：

class Solution:
    def sumEvenAfterQueries(self, nums: List[int], qs: List[List[int]]) -> List[int]:
        m, t = 0, sum(x for x in nums if x % 2 == 0)
        ans = []
        for val, idx in qs:
            if nums[idx] % 2 == 0:
                t -= nums[idx]
            nums[idx] += val
            if nums[idx] % 2 == 0:
                t += nums[idx]
            ans.append(t)
        return ans

Go 代码：

func sumEvenAfterQueries(nums []int, qs [][]int) []int {
    m, t := len(qs), 0
    ans := make([]int, m)
    for _, x := range nums {
        if x % 2 == 0 {
            t += x
        }
    }
    for i := 0; i < m; i++ {
        val, idx := qs[i][0], qs[i][1]
        if nums[idx] % 2 == 0 {
            t -= nums[idx]
        }
        nums[idx] += val
        if nums[idx] % 2 == 0 {
            t += nums[idx]
        }
        ans[i] = t
    }
    return ans
}

TypeScript 代码：

function sumEvenAfterQueries(nums: number[], qs: number[][]): number[] {
    let m = qs.length, t = 0
    for (const x of nums) t += x % 2 == 0 ? x : 0
    const ans = new Array(m)
    for (let i = 0; i < m; i++) {
        const val = qs[i][0], idx = qs[i][1]
        if (nums[idx] % 2 == 0) t -= nums[idx]
        nums[idx] += val
        if (nums[idx] % 2 == 0) t += nums[idx]
        ans[i] = t;
    }
    return ans
};

• 时间复杂度：$O(n)$
• 空间复杂度：$O(1)$

---

最后

这是我们「刷穿 LeetCode」系列文章的第 No.985 篇，系列开始于 2021/01/01，截止于起始日 LeetCode 上共有 1916 道题目，部分是有锁题，我们将先把所有不带锁的题目刷完。

在这个系列文章里面，除了讲解解题思路以外，还会尽可能给出最为简洁的代码。如果涉及通解还会相应的代码模板。

为了方便各位同学能够电脑上进行调试和提交代码，我建立了相关的仓库：https://github.com/SharingSource/LogicStack-LeetCode 。

在仓库地址里，你可以看到系列文章的题解链接、系列文章的相应代码、LeetCode 原题链接和其他优选题解。