问题描述
给出一个整数数组 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sum-of-even-numbers-after-queries
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
执行结果
代码描述
思路:先把A中偶数进行求和,然后对应着queries的index与val,去A中查找,需要分A中找到的val 与queries中的val的类型:偶数+偶数,偶数+奇数,奇数+偶数,奇数+奇数,的情况。且奇数的时候,需要注意-3%2=-1,所以加上abs().
A偶数+queries偶数,sum+=queries.val,A.val += queries.val;
A奇数+queries奇数,sum+=queries.val, sum += A.val,A.val += queries.val;
A奇数+queries偶数,sum 不变,A.val + queries.val;
A偶数+queries奇数,sum -= A.val, A.val + queries.val;
时间复杂度O(n), 空间复杂度O(n).
class Solution {
public:
vector<int> sumEvenAfterQueries(vector<int>& A, vector<vector<int>>& queries) {
vector<int> res;
int val = 0;
int index = 0;
int tempSum = sum(A); // 求A中所有偶数
for(int i = 0; i < queries.size(); ++i)
{
val = queries[i][0];
index = queries[i][1];
if(A[index] % 2 == 0 && val % 2 == 0) // 偶数+偶数
{
A[index] += val;
tempSum += val;
res.push_back(tempSum);
}
else if(abs(A[index] % 2) == 1 && abs(val % 2) == 1) // 奇数+奇数
{
tempSum += A[index];
tempSum += val;
A[index] += val;
res.push_back(tempSum);
}
else if(A[index] % 2 == 0 && abs(val % 2) == 1) // 偶数+奇数
{
tempSum -= A[index]; // 先减去原来的,再对A进行更改
A[index] += val;
res.push_back(tempSum);
}
else
{
A[index] += val;
res.push_back(tempSum); // 奇数+偶数
}
}
return res;
}
int sum(vector<int> A)
{
int result = 0;
for(int i = 0; i < A.size(); ++i)
{
if(A[i]%2 == 0)
result += A[i];
}
return result;
}
};