给出一个整数数组 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题解:
挺绕的一道小题,有几个小细节需要注意:
1、偶数和最好先求出来,然后再按照情况增加或者减少,如果是先对数组进行操作再去求和,时间复杂度会高
2、由于我们算的sum 是所有的偶数的和并不是奇数,所以当原来的数是偶数的时候,需要先将sum - 当前偶数的值,然后对数组进行赋值操作,在判断赋值后是不是偶数,如果是,sum就加上,如果不是就继续循环。
3、具体分析看代码吧
class Solution {
public int[] sumEvenAfterQueries(int[] nums, int[][] queries) {
int[] ans = new int[nums.length];//新数组,长度为nums的长度
int sum = 0;//偶数和
for(int i : nums) {
if(i % 2 == 0) {
sum += i;
}
}
for(int i = 0; i < nums.length; i++) {//遍历数组nums
int val = queries[i][0];//依题意,取出这两个数
int index = queries[i][1];
if(nums[index] % 2 == 0) {//如果添加前以前是偶数,先让 sum 减去这个数
sum -= nums[index];
}
nums[index] += val;//对数组进行添加操作
if(nums[index] % 2 == 0) {//如果添加后是偶数,就当 sum 再加回来
sum += nums[index];
}
//举个例子,以前是2,总和是4,添加6,先让4 - 2 然后再让(4 - 2) + 6即可,如果以前是奇数就不用管,因为sum是偶数和又不是奇数和
ans[i] = sum;
}
return ans;
}
}