蓝桥杯 算法提高VIP-打水问题

想说的话

大家好🌼🌼，我是 @愿此后再无WA，可以叫我小A，也可以叫我愿愿💡💡，一位阳光帅小伙，对算法领域比较感兴趣。如果我的文章对您有用，欢迎持续关注，我们一起进步！🎈🎈

很抱歉各位😪😪，现离蓝桥杯比赛不到一个月时间，我临时改变了计划，转为全心备战蓝桥，因为这个省一对我来说太重要了，也是我最后一次机会，我一定要拿到手📌📌，那么这样的话我在博客上花的时间就会少了很多，也将导致博客文章质量明显下降，在此我给大家说声抱歉💥💥

---

🌟🌟这些日子我真的很开心，博客上能遇到一群志同道合的兄弟姐妹真的很幸福，没有你们的支持与鼓励我早就坚持不下去了，因为有你们我才能走的更远☀️☀️熬过这段时期我一定会回来的，爱你们❤️❤️

打水问题

分析

这题一开始我想复杂了，就题目样例而言，我想的做法是1,2,5，4,7，3，6。总等待时间是4+4+3也是11。然后我就往其他方面思考问题了，我竟然还往均值那边去想了，越扯越远，幸好不是比赛，是比赛就完蛋。

回神后发现，其实就是条条大路通罗马，不止一种打水方式是最优的。那么我们可以使用贪心的思想谁用的时间短就谁先，当对时间序列排序后假设有两个水龙头，第一个人必定是第一个，第二个人必定是第二个，第三个人必定是第一个，第四个人必定是第二个等等…找到规律了吗？没错就是这样，因为是排好序了，所以每次轮到的人都是接着前面最先接完水的那个人，即，每队的间隔已经内定，既然内定了，所以我们就能一遍过数组，后一个人的等待时间等于前一个人的等待时间加上前一个人的接水时间，同时统计总的接水时间。需要注意的是，每个水龙头最后一个接水的人接水的时间是不用计算的，他只需要计算前一个人接水的时间加上前一个人等待的时间即可。

代码

N, M = map(int,input().split())

nums = list(map(int,input().split()))

nums.sort()

cnt = 0

for i in range(N-M):

    nums[i+M] += nums[i]

    cnt += nums[i]

print(cnt)