题目如下:
解题思路:题目很简单,有一点需要注意,就是每次执行query后,不需要遍历整个数组求出所有偶数的值。只需要记录上一次偶数的和,执行query前,如果该值是偶数,用上一次的和减去该值;执行query后,如果新值是偶数,用上一次的和加上这个新值,就可以得到这次query执行后的偶数总和。
代码如下:
class Solution(object):
def sumEvenAfterQueries(self, A, queries):
"""
:type A: List[int]
:type queries: List[List[int]]
:rtype: List[int]
"""
res = []
count = None
for val,inx in queries:
before = A[inx]
A[inx] += val
after = A[inx]
if count == None:
count = sum(filter(lambda x: x % 2 == 0,A))
else:
if before % 2 == 0:
count -= before
if after % 2 == 0:
count += after
res.append(count)
return res