蓝桥杯算法提高VIP-打水问题-题解(C++代码)

题目描述：
N个人要打水，有M个水龙头，第i个人打水所需时间为Ti，请安排一个合理的方案使得所有人的等待时间之和尽量小。

提示：
一种最佳打水方案是，将N个人按照Ti从小到大的顺序依次分配到M个龙头打水。
例如样例中，Ti从小到大排序为1，2，3，4，5，6，7，将他们依次分配到3个龙头，则去龙头一打水的为1，4，7；去龙头二打水的为2,5；去第三个龙头打水的为3,6。
第一个龙头打水的人总等待时间 = 0 + 1 + (1 + 4) = 6
第二个龙头打水的人总等待时间 = 0 + 2 = 2
第三个龙头打水的人总等待时间 = 0 + 3 = 3
所以总的等待时间 = 6 + 2 + 3 = 11

输入：
第一行两个正整数N M 接下来一行N个正整数Ti。
N,M< =1000，Ti< =1000
输出
最小的等待时间之和。（不需要输出具体的安排方案）

样例输入：
7 3
3 6 1 4 2 5 7

样例输出：
11

解题思路：
在本题的题目描述中，已经把问题解释得很清楚了，按照提示一步一步来写代码就可以很快解出来了。
（1）本题没有要求输出具体安排，如果要输出具体安排。队列是一个很好的选择。先创建m个队列数组，然后把分配的人放进去，在输出方案的同时，进行等待时间的统计。但是本题不需要用到数组
（2）先把等待的时间按照从小到大的顺序排列，用sort排序方便很多，sort函数在头文件#include<algorithm>中。（sort函数超级好用，如果有不太了解的小可爱建议百度一下）
（3）被分为一个水龙头的人的下标，相当于一个等差数组，就是i+mk，k从1开始累加，直到(i+mk)>=n。第k个人的等待时间=前面人的等待时间+打水时间。（我在这里没有改变打水时间的值，但是我看到有人是一边算等待时间一边把打水时间改成等待时间的，我觉得这样很厉害，但是我想不到，学到了，学到了）
（4）在完成一个水龙头的等待时间和之后，在加到总时间里面去。

参考代码：

#include<iostream>
#include<algorithm>
using namespace std;
int main()
{
    int n,m,i,ans=0;
    int p[1005];
    cin>>n>>m;
    for(i=0;i<n;i++)
        cin>>p[i];
    sort(p,p+i);
    for(i=0;i<m;i++)
    {
        int k=1;
        int sum=0;
        while(1)
        {
            if((i+m*k)>=n)  break;
            for(int j=k-1;j>=0;j--)
                sum+=p[i+m*(j)];
            k++;
        }
        ans+=sum;
    }
    cout<<ans<<endl;
    return 0;
}