Leetcode 985.查询后的偶数和
1 题目描述(Leetcode题目链接)
给出一个整数数组 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
2 题解
先计算所有偶数的和s,然后遍历查询数组对数组进行操作。
- 如果查询的位置是偶数则 s = s − A [ i ] s=s-A[i] s=s−A[i]
- 然后再判断 A [ i ] + v a l A[i]+val A[i]+val是否为偶数,如果是则 s = s + A [ i ] + v a l s=s+A[i]+val s=s+A[i]+val
class Solution:
def sumEvenAfterQueries(self, A: List[int], queries: List[List[int]]) -> List[int]:
res = []
s = 0
for num in A:
if num % 2 == 0: s += num
for val, index in queries:
if A[index]%2 == 0:
s -= A[index]
if (A[index] + val) % 2 == 0:
s += A[index] + val
A[index] += val
res.append(s)
return res