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

打水问题

问题描述：
　　N个人要打水，有M个水龙头，第i个人打水所需时间为Ti，请安排一个合理的方案使得所有人的等待时间之和尽量小。
输入格式
　　第一行两个正整数N M 接下来一行N个正整数Ti。
　　N,M< =1000，Ti< =1000
输出格式
　　最小的等待时间之和。（不需要输出具体的安排方案）
样例输入
7 3
3 6 1 4 2 5 7
样例输出
11
提示
　　一种最佳打水方案是，将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

解题思路：b数组记录每个水龙头共有几个人打水

会花解题思路：
#include<iostream>
#include"stdio.h"
#include<cstring>
#include<algorithm>
using namespace std;
int cmp(int x,int y)
{
    return x<y;
}
int main()
{
    int n,m;
    int a[1005]={0};
    int b[1005]={-1};
	while(scanf("%d %d",&n,&m)!=EOF){
	int sum=0;
	for(int i=0;i<n;i++)
    {
        scanf("%d",&a[i]);
    }
    sort(a,a+n,cmp);
    memset(b,-1,sizeof(b));
    for(int i=0;i<n;i++)
    {
        b[i%m]++;
    }
    for(int i=0;i<n;i++)
    {
        if(b[i%m]>0)
        {
            sum+=a[i]*(b[i%m]);
           //printf("%d*%d\n",a[i],(b[i%m]));
            b[i%m]--;
        }
    }
    printf("%d\n",sum);
	}
	return 0;
}