题目:
给出一个整数数组 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]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sum-of-even-numbers-after-queries
遇到问题:
如果使用单纯的暴力解法,即每次查询queries都遍历A数组将其中的偶数求和,当queries中数量变多时会超出事件限制。
优化思想:
由于每次查询都会改变A数组中的一个值,所以可以不用每次都遍历A数组,只需要对上一次求得的偶数和做出增减即可。
实现步骤:
一、首先求得原始A的偶数和sum;
二、遍历查询queries;
三、将查询到的val后判断加判断:
- 如果增加前A[index]为偶数,则有两种情况:第一、增加后为偶数,所以只需要将sum+val即为此次查询后的偶数和;第二、增加后为奇数,因为是奇数,所以此次查询时候将不加算该A[index],所以只需要将原来sum-A[index],并改变查询后的A[index].
- 如果增加前的A[index]为奇数,则有两种情况:第一、增加后为偶数,所以只需要将sum加上增加后的A[index];第二、增加后为奇数,sum将不做变化
四、将每一次查询后的sum添加到answer中
注意:每次查询后A都会永久改变,所以每次查询都要加val
代码:
public int[] sumEvenAfterQueries(int[] A, int[][] queries) {
int answer[] = new int[queries.length];
int val; //值
int index; //索引
int sum = 0; //偶数和
for(int j=0; j<A.length; j++){ //遍历原数组,得到原始的偶数和
if(A[j]%2 == 0){
sum += A[j];
}
}
for(int i=0; i<queries.length; i++){ //遍历查询
val = queries[i][0];
index = queries[i][1];
if(A[index] % 2 == 0){ //原来为偶数
if((A[index] + val) % 2 == 0){ //增加后任然为偶数
A[index] += val;
sum += val;
}
else{ //增加后为奇数
sum -= A[index];
A[index] += val;
}
}else{ //原来为奇数
if((A[index] + val) % 2 == 0){ //增加后为偶数
A[index] += val;
sum += A[index];
}
else{ //增加后任然为奇数
A[index] += val;
}
}
answer[i] = sum; //向answer中添加每次查询偶数和
}
return answer;
}
代码略显繁琐,可优化