赛码网 Manager 最后录取多少人

题目链接

https://www.acmcoder.com/#/practice/code

思路

用nums数组记录a中每个元素的个数，is_choose数组记录当前的人数是否被选择过；
首先对数组a从大到小进行排序；
然后遍历数组a的元素i，若当前元素的个数nums[i] > 0，且is_choose[i] = 0，那么就可以选取这个元素，因为是从大到小进行遍历，所以这个元素可以依次替代比它小的元素，可以替代的数量即为nums[i]，这里使用等差数列求和公式就可以计算出来，然后修改is_choose数组；
若is_choose[i] = 1，虽然说明已经选取过了这个元素，但我们并不能跳过它，因为这个元素可以替代比它小的还未被选取的值，我们找到那个最大的还未被选取的值，计算过程与上一步相同。

Python代码

k = int(input())
a = [int(i) for i in input().split()]
nums = [0] * 100005
is_choose = [0] * 100005
a = sorted(a, reverse=True)
res = 0

for i in range(len(a)):
    nums[a[i]] += 1

for i in a:
    if nums[i] > 0:
        j = i
        while is_choose[j] == 1 and j >= 1:
            j -= 1
        if j == 0:
            # 若j已经等于0了，说明已经无法再选取元素了，直接break
            break
        start = j - nums[i] + 1
        tmp = (start + j) * nums[i] // 2
        res += tmp
        is_choose[start:j + 1] = [1] * nums[i]
        nums[i] = 0
        
print(res)

结果

只通过了67%，不知道问题出在哪里。。。
有懂的大神可以说下思路qwq