-
Notifications
You must be signed in to change notification settings - Fork 1
/
SumOfEvenNumbersAfterQueries.java
73 lines (63 loc) · 2.62 KB
/
SumOfEvenNumbersAfterQueries.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
package com.hncboy;
import java.util.Arrays;
/**
* @author hncboy
* @date 2019/12/14 14:53
* @description 985.查询后的偶数和
*
* 给出一个整数数组 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
*/
public class SumOfEvenNumbersAfterQueries {
public static void main(String[] args) {
SumOfEvenNumbersAfterQueries s = new SumOfEvenNumbersAfterQueries();
int[] A = {1, 2, 3, 4};
int[][] queries = {{1, 0}, {-3, 1}, {-4, 0}, {2, 3}};
System.out.println(Arrays.toString(s.sumEvenAfterQueries(A, queries)));
}
public int[] sumEvenAfterQueries(int[] A, int[][] queries) {
// 计算当前偶数和
int currentEvenSum = 0;
for (int a : A) {
if (Math.abs(a) % 2 == 0) {
currentEvenSum += a;
}
}
int[] answer = new int[queries.length];
for (int i = 0; i < queries.length; i++) {
// 如果 A[queries[i][1]] 是偶数,则 currentEvenSum-A[queries[i][1]]
if (Math.abs(A[queries[i][1]]) % 2 == 0) {
currentEvenSum -= A[queries[i][1]];
}
// 修改A[queries[i][1]]
A[queries[i][1]] += queries[i][0];
// 如果修改后的 A[queries[i][1]]是偶数,则 currentEvenSum+A[queries[i][1]];
if (Math.abs(A[queries[i][1]]) % 2 == 0) {
currentEvenSum += A[queries[i][1]];
}
// 写入更新后的 currentEvenSum
answer[i] = currentEvenSum;
}
return answer;
}
}