LeetCode 985. 查询后的偶数和 Sum of Eve

作者: 1江春水 | 来源:发表于2019-07-30 19:51 被阅读3次

    【题目描述】
    给出一个整数数组 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】
    1、开始的思路是循环遍历queries 得出最终的数组A,然后遍历数组A,是偶数的相加得到最后的结果,时间复杂度太高啦!
    2、时间复杂度O(n^2)
    3、代码随便看下吧

    func sumEvenAfterQueries(_ A: [Int], _ queries: [[Int]]) -> [Int] {
        var res = [Int]()
        var tmpArr = A
        for i in 0..<queries.count {
            let index = queries[i][1]
            let value = queries[i][0]
            tmpArr[index]+=value
            var sum = 0
            for num in tmpArr {
                if num%2 == 0 {
                    sum+=num
                }
            }
            res.append(sum)
        }
        return res
    }
    

    【思路2】
    1、因为刚开始数组A内部的元素是不变的,可以先求出原始数组A的偶数和
    2、后边每遍历一次queries,只改变数组A中的一个元素
    3、所以只要判断A[index]这个值 前后变化就可以
    4、如果在赋值新值前, A[index]是偶数,需要先减去A[index] (多读几遍就懂了,😝)
    5、然后A[index]+=value后,如果是偶数,sum+=A[index]
    6、时间复杂度O(n+k),n为数组A.count k为queries.count
    7、空间复杂度O(n)

    Swift代码实现

    func sumEvenAfterQueries(_ A: [Int], _ queries: [[Int]]) -> [Int] {
       if A.count == 1 && A.first!%2 != 0 {
           return [0]
       }
       var res = [Int]()
       var tmpArr = A
       var sum = 0
       for num in tmpArr {
           if num%2 == 0 {
               sum+=num//最开始的偶数和
           }
       }
       for i in 0..<queries.count {
           let index = queries[i][1]
           let value = queries[i][0]
           if tmpArr[index]%2 == 0 {//没相加之前 
               sum-=tmpArr[index]
           }
           tmpArr[index]+=value
           if tmpArr[index]%2 == 0 {//相加之后
               sum+=tmpArr[index]
           }
           res.append(sum)
       }
       return res
    }
    

    相关文章

      网友评论

        本文标题:LeetCode 985. 查询后的偶数和 Sum of Eve

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