LeetcodeT985

2020-03-17  本文已影响0人  C调路过
//[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。
// 
//
// 
//
// 提示: 
//
// 
// 1 <= A.length <= 10000 
// -10000 <= A[i] <= 10000 
// 1 <= queries.length <= 10000 
// -10000 <= queries[i][0] <= 10000 
// 0 <= queries[i][1] < A.length 
// 
// Related Topics 数组


public class T985 {

    public static void main(String[] args) {
        System.out.println(new T985()
                .sumEvenAfterQueries(new int[]{1, 2, 3, 4},
                        new int[][]{{1, 0}, {-3, 1}, {-4, 0}, {2, 3}}));
    }

    public int[] sumEvenAfterQueries(int[] A, int[][] queries) {
        int sum = 0;
        int[] ans = new int[A.length];
        // 第一遍遍历先求不操作的和
        for (int i = 0; i < A.length; i++) {
            //或用位运算 A[i] &1 来代替
            if (A[i] % 2 == 0) {
                sum += A[i];
            }
        }

        for (int i = 0; i < queries.length; i++) {
            int val = queries[i][0];
            int index = queries[i][1];
            int t = A[index];

            if (t % 2 == 0) {
                if (val % 2 == 0) {//原本是偶数,现在加上个偶数
                    sum += val;
                } else {//原本是偶数,现在加上个奇数
                    sum -= t;
                }
            } else if (t % 2 != 0) {
                if (val % 2 == 0) {//原本是奇数,现在加上个偶数
                    // 总和不变
                } else {//原本是奇数,现在加上个奇数
                    sum += (t + val);
                }
            }
            A[index] += val;
            ans[i] = sum;
        }
        return ans;

    }

}

上一篇下一篇

猜你喜欢

热点阅读