要求
给出一个整数数组 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。
核心代码
class Solution:
def sumEvenAfterQueries(self, nums: List[int], queries: List[List[int]]) -> List[int]:
sumofEven = sum(num for num in nums if not num % 2)
result = []
for item in queries:
value = item[0]
index = item[1]
newvalue = nums[index] + value
if not nums[index] % 2 and not newvalue % 2:
sumofEven += value
elif not nums[index] % 2 and newvalue % 2:
sumofEven -= nums[index]
elif nums[index] % 2 and not newvalue % 2:
sumofEven += newvalue
result.append(sumofEven)
nums[index] = newvalue
return result
解题思路:我们先计算这个列表中所有的偶数的和,然后分三种情况,本身是偶数后面还是偶数,只要加上增量就可以了,本身是偶数,后面变成了奇数,将原本在那个位置上的值摘除掉即可,原本是奇数,后来是偶数,需要将新生成的值加入其中,前面的变化会影响到后面的数据,需要将变动更新到列表中。