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. 1 <= A.length <= 10000
2. -10000 <= A[i] <= 10000
3. 1 <= queries.length <= 10000
4. -10000 <= queries[i][0] <= 10000
5. 0 <= queries[i][1] < A.length

---

二、题解思路

---

• 题解思路：先进原始偶数总和的计算，接下来每步只要针对修改的元素在上一步总和进行加减操作即可；在后续加减上分为四个情况判断（奇数->偶数、奇数->奇数、偶数->奇数、偶数->偶数）。

---

三、代码实现

---

• C++代码实现

class Solution {
public:
    vector<int> sumEvenAfterQueries(vector<int>& A, vector<vector<int>>& queries) 
    {
        int size_1 = A.size();   //查询次数
        vector<int> nums;        //最后的输出数组
        int sum = 0;             //每一步查询之后的偶数元素的总和
        for(int i = 0;i<size_1;i++)   //先进原始偶数总和的计算，接下来每步只要针对修改的元素在上一步总和进行加减操作即可
        {
            if(A[i]%2==0)
                sum += A[i];
        }
        for(int i = 0;i<size_1;i++)
        {
            int temp = A[queries[i][1]];   //存原来的元素
            A[queries[i][1]] = A[queries[i][1]]+queries[i][0];     //对A上index位置上的元素加上val
            if(temp % 2 != 0 && A[queries[i][1]] % 2 == 0)         //奇数变偶数
                sum += A[queries[i][1]];                           //上一步总和上叠加此偶数元素                 
            else if(temp % 2 == 0 && A[queries[i][1]] % 2 != 0)    //偶数变奇数
                sum -= temp;                                       //上一步总和上减去上一步此位置上的偶数元素
            else if(temp % 2 == 0 && A[queries[i][1]] % 2 == 0)    //偶数变成偶数 
                sum = sum - temp + A[queries[i][1]]; //上一步总和上先减去上一步此位置上的偶数元素再加上当前步位置上的偶数元素
            else                                                    //奇数变成奇数
                ;    //空语句，总和不变
            nums.push_back(sum);
        }
        return nums;
    }
};