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。

 

解题思路：

  先对A数组的偶数值进行求和，然后根据变化之后的情况对已经求得的偶数之和进行修改，具体情况如下：

1. 修改之前是奇数，修改之后变成偶数，总的和要加上修改之后的偶数，并更新修改之后的值
2. 修改之前是奇数，修改之后还是奇数，则只要更新修改之后的值即可
3. 修改之前是偶数，修改之后还是偶数，总的和要加上变化的数值即val，并更新修改之后的值
4. 修改之前是偶数，修改之后变成奇数，总的和要减去修改前的A[index]，并更新修改之后的值

 

代码实现：

def sumEvenAfterQueries(A, queries):
    answer = []
    sm = sum(i for i in A if i % 2 == 0) # 先求出总的偶数和
    for x in range(len(queries)):
        val = queries[x][0]
        index = queries[x][1]
        if A[index] % 2 != 0: # 对应第1种情况
            if (A[index] + val) % 2 == 0:
                sm += A[index] + val
                A[index] += val
            else: # 对应第2种情况
                A[index] += val
        else:
            if (A[index] + val) % 2 == 0: # 对应第3种情况
                sm += val
                A[index] += val
            else: # 对应第4种情况
                sm -= A[index]
                A[index] += val
        answer.append(sm)
    return answer

 

 

                                              关注我的公众号免费领取  30G python学习资料！！！！