3080. Mark Elements on Array by Performing Queries
Description
You are given a 0-indexed array nums
of size n
consisting of positive integers.
You are also given a 2D array queries
of size m
where queries[i] = [indexi, ki]
.
Initially all elements of the array are unmarked.
You need to apply m
queries on the array in order, where on the ith
query you do the following:
- Mark the element at index
indexi
if it is not already marked. - Then mark
ki
unmarked elements in the array with the smallest values. If multiple such elements exist, mark the ones with the smallest indices. And if less thanki
unmarked elements exist, then mark all of them.
Return an array answer of size m
where answer[i]
is the sum of unmarked elements in the array after the ith
query.
Example 1:
Input: nums = [1,2,2,1,2,3,1], queries = [[1,2],[3,3],[4,2]]
Output: [8,3,0]
Explanation:
We do the following queries on the array:
- Mark the element at index
1
, and2
of the smallest unmarked elements with the smallest indices if they exist, the marked elements now arenums = [1,2,2,1,2,3,1]
. The sum of unmarked elements is2 + 2 + 3 + 1 = 8
. - Mark the element at index
3
, since it is already marked we skip it. Then we mark3
of the smallest unmarked elements with the smallest indices, the marked elements now arenums = [1,2,2,1,2,3,1]
. The sum of unmarked elements is3
. - Mark the element at index
4
, since it is already marked we skip it. Then we mark2
of the smallest unmarked elements with the smallest indices if they exist, the marked elements now arenums = [1,2,2,1,2,3,1]
. The sum of unmarked elements is0
.
Example 2:
Input: nums = [1,4,2,3], queries = [[0,1]]
Output: [7]
Explanation: We do one query which is mark the element at index 0
and mark the smallest element among unmarked elements. The marked elements will be nums = [1,4,2,3]
, and the sum of unmarked elements is 4 + 3 = 7
.
Constraints:
n == nums.length
m == queries.length
1 <= m <= n <= 105
1 <= nums[i] <= 105
queries[i].length == 2
0 <= indexi, ki <= n - 1
Solutions
Solution 1: Sorting + Simulation
First, we calculate the sum $s$ of the array $nums$. We define an array $mark$ to indicate whether the elements in the array have been marked, initializing all elements as unmarked.
Then, we create an array $arr$, where each element is a tuple $(x, i)$, indicating that the $i$-th element in the array has a value of $x$. We sort the array $arr$ by the value of the elements. If the values are equal, we sort them in ascending order of the index.
Next, we traverse the array $queries$. For each query $[index, k]$, we first check whether the element at index $index$ has been marked. If it has not been marked, we mark it and subtract the value of the element at index $index$ from $s$. Then, we traverse the array $arr$. For each element $(x, i)$, if element $i$ has not been marked, we mark it and subtract the value $x$ corresponding to element $i$ from $s$, until $k$ is $0$ or the array $arr$ is fully traversed. Then, we add $s$ to the answer array.
After traversing all the queries, we get the answer array.
The time complexity is $O(n \times \log n)$, and the space complexity is $O(n)$. Here, $n$ is the length of the array $nums$.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
|