985. Sum of Even Numbers After Queries
Problem Description
In this problem, you are provided with two pieces of information: 1) an integer array nums
, which contains some integers, and 2) an array of arrays queries
, where each subarray contains exactly two integers representing a value (val_i
) and an index (index_i
). The goal is to process each subarray in queries
sequentially. The processing steps for each query are as follows:
a) Add val_i
to nums[index_i]
(modify the element in nums
at the specified index by adding the value).
b) Calculate the sum of all even numbers currently in the nums
array.
You need to perform this two-step process for each query and record the result of the second step. The final output should be an array where each element corresponds to the sum of even numbers after the corresponding query has been processed.
Intuition
The naive approach to solving this problem would be to compute the sum of all even numbers in the nums
array after applying each query. However, this would be inefficient because it would require iterating through the entire nums
array to recalculate the sum after each query, resulting in a time complexity that would be burdensome for large arrays.
The solution instead maintains a running total sum s
of all even numbers in the nums
array. This sum is updated incrementally. Here's the thought process for this approach:
- Calculate and store the sum of all even numbers in the
nums
array before processing any queries. - For each query
(val_i, index_i)
:- If the number at
nums[index_i]
is even before the query, subtract it froms
to temporarily remove it from the even sum. This is done because the addition might make it odd. - Apply the update
nums[index_i] = nums[index_i] + val_i
. - After the update, check if
nums[index_i]
is now even. If it is, add it back tos
, as the change either kept it even or turned it from odd to even.
- If the number at
- Record the current value of
s
after each query to build the answer array, which represents the sum of even numbers after each modification.
This way, we avoid recalculating the sum of even numbers from scratch after each query, effectively reducing complexity and speeding up the algorithm.
Solution Approach
The implementation of the solution uses an approach that efficiently updates and calculates the sum of even numbers after each query applied to the nums
array. The initial setup and steps followed in the code are as follows:
-
Initialize the Even Sum: Before processing any queries, the solution calculates the initial sum
s
of all the even numbers innums
. This is done using a list comprehension and the built-insum
function:s = sum(x for x in nums if x % 2 == 0)
-
Process Queries: The implementation then iterates over each query in
queries
. For each query[v, i]
, wherev
is the value to be added andi
is the index, the solution:- Checks if
nums[i]
is even before the update. If it is, the value ofnums[i]
is subtracted from the running even sums
, as the update might turn it into an odd number:if nums[i] % 2 == 0: s -= nums[i]
- Applies the update to
nums[i]
by adding the valuev
to it:nums[i] += v
- Checks if after the update,
nums[i]
is even. If it has become or remained even, it is added to the running even sums
:if nums[i] % 2 == 0: s += nums[i]
- Append the current running sum
s
(which now reflects the sum of even values after the update) to an answer listans
:ans.append(s)
- Checks if
-
Return Results: After processing all the queries, the list
ans
, which has recorded the sum of even numbers after each query, is returned as the final output.
By maintaining and updating the running even sum s
with each query, the algorithm achieves an efficient and dynamic way to keep track of changes without recomputing the entire sum after each update. This greatly improves the time efficiency, especially when there are many queries or the nums
array is large.
Ready to land your dream job?
Unlock your dream job with a 2-minute evaluator for a personalized learning plan!
Start EvaluatorExample Walkthrough
Let's go through a small example to illustrate the solution approach.
Suppose we are given the integer array nums = [1, 2, 3, 4]
and queries [[1, 0], [-3, 2], [5, 3], [4, 1]]
. We need to perform the queries on nums
and track the sum of even numbers in nums
after each query.
-
Initialize the Even Sum:
Before any queries, we find the sum of all even numbers in
nums
:nums = [1, 2, 3, 4] s = sum(x for x in nums if x % 2 == 0) # s = 2 + 4 = 6
The initial sum
s
of even numbers innums
is 6. -
Process Queries:
First query
[1, 0]
: We add 1 tonums[0]
, which results innums
becoming[2, 2, 3, 4]
.Since
nums[0]
was not even before the query (1
is odd), we don't subtract anything froms
. After the update,nums[0]
becomes even (2
), so we add it tos
.s += nums[0] # s = 6 + 2 = 8 ans.append(s) # ans = [8]
Second query
[-3, 2]
: We subtract 3 fromnums[2]
(nums[2] - 3
), andnums
becomes[2, 2, 0, 4]
.Before the query,
nums[2]
was odd, so the sums
remains unchanged. After the update,nums[2]
is even (0
), and it's added to the sums
.s += nums[2] # s = 8 + 0 = 8 (unchanged) ans.append(s) # ans = [8, 8]
Third query
[5, 3]
: We now add 5 tonums[3]
, resulting innums
becoming[2, 2, 0, 9]
.nums[3]
was even before (4
), so we subtract it from the sums
first. After the update (4 + 5 = 9
),nums[3]
is odd, so we don't add it back tos
.s -= nums[3] # Subtracting the old value of `nums[3]`, s = 8 - 4 = 4 nums[3] += 5 # nums[3] becomes odd after the addition ans.append(s) # ans = [8, 8, 4]
Fourth query
[4, 1]
: The last query adds 4 tonums[1]
, changingnums
to[2, 6, 0, 9]
.Since
nums[1]
was even before (2
), we remove it from the sums
. After the update (2 + 4 = 6
),nums[1]
is still even, so we add the new value back tos
.s -= nums[1] # Subtracting the initial value of `nums[1]`, s = 4 - 2 = 2 nums[1] += 4 # Updating `nums[1]` s += nums[1] # Adding the updated value to `s`, s = 2 + 6 = 8 ans.append(s) # ans = [8, 8, 4, 8]
-
Return Results:
After processing all the queries, the
ans
list, which contains the sum of even numbers after each query, is[8, 8, 4, 8]
. This list is the final output.
Thus, by following this approach, we efficiently kept track of the sum of even numbers in nums
after each query without recalculating the entire sum each time.
Solution Implementation
1from typing import List
2
3class Solution:
4 def sumEvenAfterQueries(self, nums: List[int], queries: List[List[int]]) -> List[int]:
5 # Calculate the initial sum of even numbers in the array
6 sum_even = sum(value for value in nums if value % 2 == 0)
7 result = [] # Initialize the result list
8
9 # Iterate over each query in the queries list
10 for value_to_add, index in queries:
11 # Check if the number at the index is even before the operation
12 if nums[index] % 2 == 0:
13 # If it's even, subtract it from sum_even because it might change
14 sum_even -= nums[index]
15
16 # Add the value to the number at the given index
17 nums[index] += value_to_add
18
19 # Check if the number at the index is even after the operation
20 if nums[index] % 2 == 0:
21 # If it's even now, add it to sum_even
22 sum_even += nums[index]
23
24 # Add the current sum of even numbers to the result list
25 result.append(sum_even)
26
27 # Return the result list which contains the sum of even numbers after each query
28 return result
29
1class Solution {
2 // Function to calculate the sums of even valued numbers after each query
3 public int[] sumEvenAfterQueries(int[] nums, int[][] queries) {
4 int evenSum = 0; // Variable to keep track of the sum of even numbers
5
6 // Initial pass to calculate sum of even numbers in the original array
7 for (int num : nums) {
8 if (num % 2 == 0) {
9 evenSum += num;
10 }
11 }
12
13 // The number of queries
14 int numQueries = queries.length;
15 // Array to hold results after each query
16 int[] ans = new int[numQueries];
17 // Index for placing results in the 'ans' array
18 int resultIndex = 0;
19
20 // Iterate over each query
21 for (int[] query : queries) {
22 int value = query[0]; // The value to be added
23 int index = query[1]; // The index of the nums array to which the value is to be added
24
25 // If the current number at index is even, subtract it from the evenSum
26 if (nums[index] % 2 == 0) {
27 evenSum -= nums[index];
28 }
29
30 // Update the number by adding the value from the query
31 nums[index] += value;
32
33 // If the updated number is even, add it to the evenSum
34 if (nums[index] % 2 == 0) {
35 evenSum += nums[index];
36 }
37
38 // Store the current sum of even numbers in the result array
39 ans[resultIndex++] = evenSum;
40 }
41
42 // Return the array with results after each query
43 return ans;
44 }
45}
46
1class Solution {
2public:
3 vector<int> sumEvenAfterQueries(vector<int>& nums, vector<vector<int>>& queries) {
4 int sumEven = 0; // Variable to store the sum of even numbers
5
6 // Calculate the initial sum of even elements in nums
7 for (int num : nums) {
8 if (num % 2 == 0) {
9 sumEven += num;
10 }
11 }
12
13 vector<int> result; // Vector to store the results after each query
14 result.reserve(queries.size()); // Reserve space to avoid reallocations
15
16 // Process each query
17 for (auto& query : queries) {
18 int val = query[0]; // Value to add
19 int index = query[1]; // Index at which the value is to be added
20
21 // If the original number at index is even, subtract it from sumEven
22 if (nums[index] % 2 == 0) {
23 sumEven -= nums[index];
24 }
25
26 // Add the value to the number at index
27 nums[index] += val;
28
29 // If the new number at index is even, add it to sumEven
30 if (nums[index] % 2 == 0) {
31 sumEven += nums[index];
32 }
33
34 // Append the current sumEven to the result
35 result.push_back(sumEven);
36 }
37
38 return result; // Return the final result vector
39 }
40};
41
1function sumEvenAfterQueries(nums: number[], queries: number[][]): number[] {
2 // Initialize sum of even numbers in the array.
3 let sumEven = 0;
4
5 // Calculate the initial sum of all even numbers in the 'nums' array.
6 for (const num of nums) {
7 if (num % 2 === 0) {
8 sumEven += num;
9 }
10 }
11
12 // Prepare the array to store result after each query.
13 const result: number[] = [];
14
15 // Iterate over each query in 'queries' array.
16 for (const [valueToAdd, index] of queries) {
17 // If the element at the current index is even, subtract it from 'sumEven'.
18 if (nums[index] % 2 === 0) {
19 sumEven -= nums[index];
20 }
21
22 // Add the value from the query to the element at the current index.
23 nums[index] += valueToAdd;
24
25 // If the updated element at the current index is even, add it to 'sumEven'.
26 if (nums[index] % 2 === 0) {
27 sumEven += nums[index];
28 }
29
30 // Append the current sum of even numbers to the result array.
31 result.push(sumEven);
32 }
33
34 // Return the final result array.
35 return result;
36}
37
Time and Space Complexity
The provided code snippet consists of an initial computation that sums the even values of the input nums
list, followed by an iteration through each query in queries
. Each query modifies a single element of nums
and conditionally updates the sum of even numbers.
Time Complexity:
The initial sum computation has a time complexity of O(N)
, where N
is the number of elements in nums
.
The for loop runs for every query in queries
, which we can denote as Q
, where Q
is the number of queries. Inside the loop, we have constant-time operations. Therefore, the loop runs in O(Q)
time.
The total time complexity of the function is a result of the initial sum computation and the queries loop, which we can express as O(N) + O(Q)
. Therefore, the overall time complexity is O(N + Q)
.
Space Complexity:
The space complexity is determined by the additional space used by the algorithm aside from the input itself. The space used includes the sum s
and the list ans
that accumulates the resulting sums after each query.
s
is a single integer, so it requiresO(1)
space.ans
will hold the result after every query, which means it will grow to the size ofQ
. Therefore, it requiresO(Q)
space.
Considering the additional space used by the function, the total space complexity is O(Q)
since Q
may vary independently of N
and the size of ans
directly depends on the number of queries.
Learn more about how to find time and space complexity quickly using problem constraints.
Which data structure is used to implement priority queue?
Recommended Readings
LeetCode Patterns Your Personal Dijkstra's Algorithm to Landing Your Dream Job The goal of AlgoMonster is to help you get a job in the shortest amount of time possible in a data driven way We compiled datasets of tech interview problems and broke them down by patterns This way we
Recursion Recursion is one of the most important concepts in computer science Simply speaking recursion is the process of a function calling itself Using a real life analogy imagine a scenario where you invite your friends to lunch https algomonster s3 us east 2 amazonaws com recursion jpg You first
Runtime Overview When learning about algorithms and data structures you'll frequently encounter the term time complexity This concept is fundamental in computer science and offers insights into how long an algorithm takes to complete given a certain input size What is Time Complexity Time complexity represents the amount of time
Want a Structured Path to Master System Design Too? Don’t Miss This!