2040: 打水问题（贪心+模拟)

题目描述

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

来源

算法提高

题意：看过样例加提示，应该就能明白这个题怎么写，要找出最省的，肯定是要排序的，然后对水龙头进行分配，按m个水龙头，从小大到，分配，  说不明白 举个例子  有5个水龙头   有11个人  分别为 

1 5 6 7 8 9 11 12 15 17 19   一个一个分配，从第一个水龙头到最后一个 然后在从第一个开始。

分配后：

第一个 1 9 19

第二个 5 11

第三个 6 12

第四个 7 15

第五个 8 17

就这样 在算一下就好了。

代码：

# include <stdio.h>
# include <string.h>
# include <algorithm>

using namespace std;

int main(void)
{
	int n, m, a[1001], book[1001];
	while (~ scanf("%d %d", &n, &m))
	{
		int i;
		memset(book, 0, sizeof(book));  // 标记是否读过， 初始化。
		for(i = 1; i <= n; i ++)
			scanf("%d", &a[i]);
		sort(a+1, a+1+n); // 排序
		int sum, sum1 = 0, c, d;
		for (i = 1; i <= n; i ++)
		{
			sum = 0, c = 0;
			d = i;
			while (1)  // 每一个都进入循环
			{
				if (book[d] == 1) // 如果已经读过 直接跳出
					break;
				else
					book[d] = 1;  // 否则 标记
				
				if (d + m <= n)
				{
					c += a[d];  // 加上前一个人的
					sum += c;   // 总的
					d += m;   // 改变位置
				}
				else
					break;
			}
			sum1 += sum;
		}
		printf("%d\n", sum1);
	}
	return 0;
}