题目描述
给出一个整数数组 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。
解题思路
题目比较简单,注意时间复杂度,先把数组A中的偶数计算出来,然后遍历queries的时候,变更A的值,直接在计算后的结果上进行加减操作。
程序实现
class Solution {
public int[] sumEvenAfterQueries(int[] A, int[][] queries) {
int sumA = 0;
for(int j=0;j<A.length;j++){
if(A[j]%2 == 0){
sumA+= A[j];
}
}
int[] answer = new int[queries.length];
for(int i = 0; i<queries.length; i++){
int val = queries[i][0] ;
int index = queries[i][1];
int temp = A[index];
A[index]+= val;
if(temp%2 == 0){
if(A[index]%2 == 0){
sumA = sumA-temp + A[index];
}else{
sumA = sumA-temp;
}
}else{
if(A[index]%2 == 0){
sumA = sumA + A[index];
}
}
answer[i] = sumA;
}
return answer;
}
}