乐趣区

Leetcode-PHP题解D48-985-Sum-of-Even-Numbers-After-Queries

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;
    }
}

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

退出移动版