2019-03-21 [蓝桥杯][算法提高VIP]打水问题

题目描述
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
提示
C语言在线学习平台微信号dotcpp
来源
算法提高

#include<iostream>
#include<algorithm>
using namespace std;
const int N = 1010,M = 1010;
int A[N],s[M],sum;
int main(void)
{
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>A[i];
    sort(A+1,A+1+n);
    int idx=1;
    for(int i=1;i<=n;i++)
    {
        sum+=s[idx];
        s[idx]+=A[i];
        idx++;
        if(idx==m+1) idx=1;
    }
    cout<<sum;
    return 0;
}