【kickstart round A 2019】第一题Training Python题解

一杯敬月光

于 2020-08-29 05:19:55 发布

阅读量114

点赞数

分类专栏： kickstart 文章标签：算法

本文链接：https://blog.csdn.net/sinat_32682001/article/details/108289844

版权

kickstart 专栏收录该内容

21 篇文章 0 订阅

订阅专栏

题目：
教练从N个学生中挑选P个学生，并对之训练使P个同学技能值全部相同。

解题思路：
排序后，计算前缀和

代码：

T=int(input())

for tt in range(T):
    N,P=[int(s) for s in input().split()]
    students=[int(s) for s in input().split()]
    students.sort()
    pre_sum=[0]
    for i in students:
        pre_sum.append(pre_sum[-1]+i)
    ans=float('inf')
    for i in range(N-P+1):
        loc1=i
        loc2=i+P
        rec=P*students[loc2-1]-(pre_sum[loc2]-pre_sum[loc1])
        ans=min(ans,rec)
    print('Case #{}: {}'.format(tt+1,ans))