前言
Weekly Contest 122 的 查询后的偶数和:
给出一个整数数组 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
解题思路
此题有两种方法解决,如下所示:
暴力法:按照题目描述,使用嵌套的两个 for 循环就可以解决。
双指针法:先使用一个 for 循环计算出初始的偶数之和,然后在每次修改数组 A 中的元素时,分别获取修改前后的元素的值,根据情况修改偶数之和。
实现代码
暴力法
/**
* 985. 查询后的偶数和
* @param A
* @param queries
* @return
*/
public int[] sumEvenAfterQueries(int[] A, int[][] queries) {
for(int i=0;i<queries.length;i++){
// 值
int val=queries[i][0];
// 索引
int index=queries[i][1];
// 进行特定索引的元素累加
A[index]=A[index]+val;
// 偶数之和
int evenSum=0;
for(int j=0;j<A.length;j++){
if(A[j]%2==0){
evenSum+=A[j];
}
}
result[i]=evenSum;
}
return return result;
}
双指针法
/**
* 985. 查询后的偶数和
* @param A
* @param queries
* @return
*/
public int[] sumEvenAfterQueries(int[] A, int[][] queries) {
int[] result=new int[queries.length];
// 偶数之和
int evenSum=0;
for(int a:A){
if(a%2==0){
evenSum+=a;
}
}
for(int i=0;i<queries.length;i++){
// 值
int val=queries[i][0];
// 索引
int index=queries[i][1];
// 修改前的元素
int before=A[index];
// 修改后的元素
int after=A[index]+val;
A[index]=after;
if(before%2==0 && after%2!=0){// 修改前为偶数,修改后为奇数
evenSum-=before;
}else if(before%2==0 && after%2==0){// 修改前为偶数,修改后为偶数
evenSum=evenSum-before+after;
}else if(before%2!=0 && after%2==0){// 修改前为奇数数,修改后为偶数
evenSum+=after;
}else{// 修改前为奇数,修改后为奇数
// 什么都不做
}
result[i]=evenSum;
}
return result;
}