【刷题打卡】985. 查询后的偶数和

193 阅读2分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

一、题目描述:

985. 查询后的偶数和 - 力扣(LeetCode) (leetcode-cn.com)

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

二、思路分析:

题目中提到了会改变原数组,如果最常规的思路就是for循环,每改变一次A中的值需要重新计算,时间复杂度高,既然只要偶数的和只需要在原偶数的和上做加减即可,先找到当前项如果是偶数先减去,当前项修改在判断修改后是否是偶数,如果是就加上去,总的和sum是一直在修改的

三、AC 代码:

class Solution {
    public int[] sumEvenAfterQueries(int[] A, int[][] queries) {
        int len = A.length;
        int sum = 0;
        for(int num : A){
            if(num%2 == 0){
                sum += num;
            }
        }

        int[] B = new int[len];
        int index, value;
        for(int i = 0; i < len; i++){
            value = queries[i][0];
            index = queries[i][1];

            if(A[index]%2 == 0){
                sum -= A[index];
            }
            A[index] += value;

            if(A[index]%2 == 0){
                sum += A[index];
            }
            B[i] = sum;
        }
        return B;
    }
}

范文参考:

985. c++几乎双百的模拟解法 - 查询后的偶数和 - 力扣(LeetCode) (leetcode-cn.com)