leetcode.985 查询后的偶数和 (Python 3)
给出一个整数数组 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
解题思路:
题目要求输出的结果是每次更新后的list中的偶数和,那么我们就可以先计算出原始列表的偶数和num,在根据变更的数值来计算结果。
当变更的数字是偶数的时候,那么num就减去变更的数字,变更之后的数字如果是偶数,那么num就加上变更之后的数字,就可以快速得到偶数和。
代码如下:
class Solution:
def sumEvenAfterQueries(self, A: 'List[int]', queries: 'List[List[int]]') -> 'List[int]':
num = 0
re = []
for i in range(len(A)): # 求出原始列表中偶数的和
if A[i] % 2 == 0:
num += A[i]
for i in range(len(queries)): # 根据queries来计算偶数和
if A[queries[i][1]] % 2 == 0:
num -= A[queries[i][1]]
if (A[queries[i][1]] + queries[i][0]) % 2 == 0 :
num += (A[queries[i][1]] + queries[i][0])
re.append(num)
A[queries[i][1]] = A[queries[i][1]] + queries[i][0] # 将新值赋给A列表
return re