D48 985. Sum of Even Numbers After Queries

题目链接

985. Sum of Even Numbers After Queries

题目分析

给定一个初始数组A,再给定一个二维操作数组Q

操作数组里的每一个值是一个数组。其第一个值代表要添加的数。第二个值代表需要操作的数字在数组A中的下标。

也即,需要对A[Q[1]]A[Q[0]

结果中第i个元素的结果为,当执行第i步时,数组A中偶数元素的和。

思路

这题如果每一步都array_sum的话时间开销会很大,所以采取的方案是先计算初始数组中偶数的和。

再在每一步计算的过程中,判断当前位置是否为偶数。
若为偶数,那么表明在最初已经计算过偶数和了,那么把它从偶数和中减去。

判断相加了第0个元素后,是否为偶数。是则加进偶数和中。

修改原数组A用于后面计算。

把每一步的偶数和记录下来,以便最后返回。

最终代码

<?php
class Solution {

    /**
     * @param Integer[] $A
     * @param Integer[][] $queries
     * @return Integer[]
     */
    function sumEvenAfterQueries($A, $queries) {
        $evens = array_filter($A,function($val){
            return $val%2==0;
        });
        $total = array_sum($evens);
        $sums = [];
        foreach($queries as $query){
            if($A[$query[1]]%2 == 0){
                $total -= $A[$query[1]];
            }
            if(($A[$query[1]]+$query[0])%2==0){         
                $total += $query[0] + $A[$query[1]];   
            }
            $sums[] = $total;
            $A[$query[1]] += $query[0];
        }
        return $sums;
    }
}

若觉得本文章对你有用,欢迎用爱发电资助。


skys215
42 声望26 粉丝