985.查询后的偶数和Java

985.查询后的偶数和Java

题目描述

给出一个整数数组 A 和一个查询数组 queries。

对于第 i 次查询，有 val = queries[i][0], index = queries[i][1]，把 val 加到 A[index] 上。第 i 次查询的答案是 A 中偶数值的和。

返回所有查询的答案。答案以数组给出，rst[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。

本文题来自LeetCode：https://leetcode-cn.com/problems/sum-of-even-numbers-after-queries/

思路

方法一：暴力破解。每次根据queries数组改变A中的元素就遍历一遍A数组计算偶数和。

方法二：先把整个数组的偶数和sum计算出来。再遍历queries数组，如果要修改的A中的元素为偶数则在sum中减取该元素，然后再计算删除的元素加上queries的值后对sum的影响，即如果是偶数则再给sum加上它。

算法分析

1. 时间复杂度为遍历数组queries的时间复杂度Q和A的时间复杂度N的乘积，即O(N*Q)，空间复杂度为O(N)
2. 时间复杂度为O(N+Q)，空间复杂度为O(N)

求解函数

//方法一：暴力破解
public int[] sumEvenAfterQueries(int[] A, int[][] queries) {
        int[] rst = new int[A.length];
        for (int i = 0; i < queries.length; ++i){
            int sum = 0;
            A[queries[i][1]] += queries[i][0];
            for (int j = 0; j < A.length; ++j){ //遍历A数组求偶数和
                if (A[j] % 2 == 0) sum += A[j];
            }
            rst[i] = sum;
        }
        return rst;
    }

//方法二：
public int[] sumEvenAfterQueries(int[] A, int[][] queries) {
        int sum = 0;
        int[] rst = new int[A.length];
        for (int i = 0; i < A.length; ++i) { //计算偶数和sum
            if (A[i] % 2 == 0) sum += A[i];
        }
        for (int i = 0; i < queries.length; ++i) {
        	//如果原来的元素是偶数就从sum中减掉
            if (A[queries[i][1]] % 2 == 0) {
            	sum = sum - A[queries[i][1]];
            }
            A[queries[i][1]] += queries[i][0];
            //加入的值是否对sum产生影响，如果是就加到sum中
            if (A[queries[i][1]] % 2 == 0) {
            	sum = sum + A[queries[i][1]];
            }
            rst[i] = sum;
        }
        return rst;
}



    

主函数调用

class Solution985 {
    public static void main(String[] args) {
        Solution985 sol = new Solution985();
        int[] nums = new int[]{1, 2, 3, 4};
        int[][] queries = new int[][]{{1, 0}, {-3, 1}, {-4, 0}, {2, 3}};
        int[] rst = sol.sumEvenAfterQueries(nums, queries);
        for (int i = 0; i < rst.length; ++i) {
            System.out.print(rst[i] + ", ");
        }
    }

    public int[] sumEvenAfterQueries(int[] A, int[][] queries) {...}
}

输出结果