力扣刷题记录#数组#简单#985查询后的偶数和

题目描述

给出一个整数数组 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的偶数的和sum，后续只需在此基础上进行加减，不用多次遍历。

将val 加到 A[index]上，再计算更新后的A的偶数的和有四种情况：

1. A[index]原本就是偶数，加上val仍为偶数，则更新后A的偶数和：sum加上val；
2. A[index]原本就是偶数，加上val变为奇数，则更新后A的偶数和：sum减去原来A[index]的值；
3. A[index]原本是奇数，加上val变为偶数，则更新后A的偶数和：sum加上原本的a[index]和val；
4. A[index]原本是奇数，加上val仍为奇数，则更新后A的偶数和不变，仍为sum。

class Solution(object):
    def sumEvenAfterQueries(self, A, queries):
        """
        :type A: List[int]
        :type queries: List[List[int]]
        :rtype: List[int]
        """
        
        sum_even = 0   # 偶数的和
        for num in A:
            if num%2==0:
                sum_even = sum_even + num
        
        query = len(queries)   # 查询次数
        
        ans = list()
        for i in range(0,query):
            val = queries[i][0]
            index = queries[i][1]
            
            # A[index]原本就是偶数
            if A[index]%2==0: 
                # A[index]+val仍为偶数
                if (A[index]+val)%2==0:   
                    sum_even = sum_even + val
                    ans.append(sum_even)
                # A[index]+val为奇数
                else:   
                    sum_even = sum_even - A[index]
                    ans.append(sum_even)
            
            # A[index]原本是奇数
            else: 
                # A[index]+val为偶数
                if (A[index]+val)%2==0:   
                    sum_even = sum_even + A[index] + val
                    ans.append(sum_even)
                # A[index]+val为奇数
                else:   
                    ans.append(sum_even)
                    
            # 更新A[index]的值
            A[index] = A[index]+val
                    
        return ans