赛码网 Manager 最后录取多少人_赛码网 mannger-CSDN博客

本文链接：https://blog.csdn.net/weixin_44123362/article/details/129942936

文章描述了一种编程问题的解决方案，首先对输入数组进行降序排序，然后遍历数组，利用元素个数和已选择状态选择最大值，用等差数列公式计算可替代元素总数，最终更新选择状态和结果计数。代码使用Python实现，但只有67%的测试用例通过。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接

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)