蓝桥杯 排队接水2 贪心 python

问题描述

　　有N个人排队到M个水龙头去打水，他们装满水桶的时间T1,T2……Tn为整数且各不相等，应如何安排他们的打水顺序才能使他们花费的总时间最少？

输入格式

　　第1行：两个整数n和m，n表示人的个数，m表示水龙头的个数；
　　第2行，n个数分别表示n个人装水的时间
　　数据范围：m<=n/3,n<=1000,t<3000。

输出格式

　　一个整数，表示总花费的最少时间。

样例输入

6 2
5 4 6 2 1 7

样例输出

40

n,m=map(int,input().split())
la=list(map(int,input().split()))
sum=0
la.sort()
for i in range(m):
    lc=la[i:n:m]
    count=0
    for j in lc:
        count=count+j
        sum=sum+count
print(sum)