985. 查询后的偶数和(Python)

更多精彩内容,请关注【力扣简单题】。

题目

难度:★☆☆☆☆
类型:数组

给出一个整数数组 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 次查询的答案。

提示
1 <= A.length <= 10000
-10000 <= A[i] <= 10000
1 <= queries.length <= 10000
-10000 <= queries[i][0] <= 10000
0 <= queries[i][1] < A.length

示例

输入: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。

解答

我们无需每次操作都对数组中的偶数,因为修改元素前后的偶数和是有关系的。

首先,我们计算出原始数组的偶数和even_sum,我们定义了函数is_even来判断一个输入的数字是否是偶数。

我们探究将位置i处的元素A[i]修改为val时,偶数和even_sum的变换,有四种情况:

  1. A[i]和val均为偶数,则even_sum变为even_sum+val-A[i];

  2. A[i]为偶数,val为奇数,则even_sum变为even_sum-A[i];

  3. A[i]为奇数,val为偶数,则even_sum变为even_sum+val;

  4. A[i]和val均为奇数,则even_sum不变。

注意题目中的val是只在原始位置的数字上加上val,这里的val是加完的结果,记得及时修改数组中的相应元素。

class Solution:
    def sumEvenAfterQueries(self, A, queries):
        """
        :param A: List[int]
        :param queries: List[List[int]]
        :return: List[int]
        """
        is_even = lambda n: n % 2 == 0
        even_sum = sum([n for n in A if is_even(n)])
        res = []
        for val, i in queries:
            val += A[i]
            if is_even(A[i]) and is_even(val):
                change = - A[i] + val
            elif is_even(A[i]) and not is_even(val):
                change = - A[i]
            elif not is_even(A[i]) and is_even(val):
                change = val
            else:
                change = 0
            A[i] = val
            even_sum += change
            res.append(even_sum)
        return res

如有疑问或建议,欢迎评论区留言~

你可能感兴趣的:(985. 查询后的偶数和(Python))