题目描述:
给出一个整数数组 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
解题思路:
要返回每一次查询之后的偶数和,直观上可以每一次查询后遍历数组找到偶数进行求和,但由于A和query最大都是10000,采用二重循环会超时,所以要用线性的方法去做
可以先求出原数组的偶数和count,对于每一次更新判断原数字和更新值的奇偶性
1.原数字为偶数,更新值为偶数,和还是偶数:原数字已经在count中了,只需将count加上更新值即可
2.原数字为偶数,更新值为奇数,和还是奇数:原数字已经在count中了,需要将原数字从count中减掉
3.原数字为奇数,更新值为奇数,和还是偶数:原数字不在count中了,只需将count加上更新值以及原数字即可
4.原数字为奇数,更新值为偶数,和还是奇数:查询前后都是奇数,不会出现在count中
每次查询后更新数组,并将本次的count添加到返回值中即可
class Solution:
def sumEvenAfterQueries(self, A: List[int], queries: List[List[int]]) -> List[int]:
count=0
#先求出A中的偶数和
for i in A:
if i%2==0:
count+=i
res=[]
for query in queries:
if query[0]%2==0 and A[query[1]]%2==0:#原来是偶数,加偶数
count+=query[0]
elif query[0]%2!=0 and A[query[1]]%2==0:#原来是偶数,加奇数
count-=A[query[1]]
elif query[0]%2!=0 and A[query[1]]%2!=0:#原来是奇数,加奇数
count+=A[query[1]]+query[0]
# elif query[0]%2!=0 and A[query[1]]%2==0:#原来是奇数,加偶数
# count-=A[query[1]]
A[query[1]]+=query[0]#更新数组
res.append(count)
return res