美文网首页
985. 查询后的偶数和

985. 查询后的偶数和

作者: 好吃红薯 | 来源:发表于2019-05-08 13:45 被阅读0次

    给出一个整数数组 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

    思路:先计算一个sum,后面根据每个数字改动,再调整sum

    class Solution(object):
        def sumEvenAfterQueries(self, A, queries):
            """
            :type A: List[int]
            :type queries: List[List[int]]
            :rtype: List[int]
            """
            res =[]
            sum = 0
            for j in range(len(A)):
                if(A[j]%2 == 0):
                    sum = sum + A[j]
                    
            for i in range(len(queries)):
                if A[queries[i][1]]%2==0:
                    sum = sum -A[queries[i][1]]
                
                A[queries[i][1]] = A[queries[i][1]] + queries[i][0]
                if A[queries[i][1]]%2==0:
                    sum = sum +A[queries[i][1]]
                res.append(sum)
            
            return res
    

    版本二:(代码简洁修改)

    class Solution(object):
        def sumEvenAfterQueries(self, A, queries):
            """
            :type A: List[int]
            :type queries: List[List[int]]
            :rtype: List[int]
            """
            res =[]
            Sum = sum(x for x in A if x%2==0)
                    
            for k,j in queries:
                if A[j]%2==0:
                    Sum -= A[j]
                A[j] += k
                if A[j]%2==0:
                    Sum += A[j]
                res.append(Sum)
            
            return res
    

    相关文章

      网友评论

          本文标题:985. 查询后的偶数和

          本文链接:https://www.haomeiwen.com/subject/epemoqtx.html