• Leetcode 3080. Mark Elements on Array by Performing Queries
  • 1. 解题思路
  • 2. 代码实现

• 题目链接：3080. Mark Elements on Array by Performing Queries

1. 解题思路

这一题我们只需要按照题意进行一下实现就行了。具体来说的话，我们只需要依序遍历一下query，然后在每个query过程中判断一下目标元素之前是否有被mark过，并额外找出最小的$k_i$个元素进行mark即可。

因此，我们要做的就是如何最优地完成这两个操作，前者我们用一个hash table来记录下所有被mark的位置，后者我们用一个有序数列来对其进行维护即可。

2. 代码实现

给出python代码实现如下：

class Solution:def unmarkedSumArray(self, nums: List[int], queries: List[List[int]]) -> List[int]:marked = set()q = sorted([(x, i) for i, x in enumerate(nums)])s = sum(nums)ans = []for idx, k in queries:if idx not in marked:s -= nums[idx]q.pop(bisect.bisect_left(q, (nums[idx], idx)))marked.add(idx)for x, i in q[:k]:s -= xmarked.add(i)q = q[k:]ans.append(s)return ans

提交代码评测得到：耗时1076ms，占用内存50.3MB。