贪心算法——排队打水问题

*/6.3 排队打水问题：有n个人排队到r个水龙头去打水，他们装满水桶的时间为t1,t2,…,tn为正整数且个不相等。
应如何安排他们打水顺序才能使他们花费的时间最少。
算法分析:时间总和=等待时间+装水时间。采用贪心思想。先sort()，默认将装水时间从小到大排序。
n个人，r个水龙头，可以以水龙头为序将队伍序分割成n/r个小组，每小组的成员分别对应r个位置。
实际上就是将人依序分成r个小队打水。比如有三个水龙头，分成三个小队，1、4、7… ，2、5、8…，3、6、9…
可以验证是否符合贪心：从序号第1的人开始依次往后选择水龙头，为了等待时间+灌水时间最小是不是该这样站队？

*/
#include
#include
#include //C++标准模板库STL，用到sort排序
using namespace std;

int main()
{

int a[1000],b[1000],n,r; //a数组存放每个人的装水时间，b数组存放每个人的总时间（等待时间+打水时间） 
memset(a,0,sizeof(a));  //数组初始化为0 
memset(b,0,sizeof(b));
scanf("%d%d",&n,&r);	

int i,sum=0;
for(i=1; i<=n; i++)
	scanf("%d",&a[i]);
sort(a+1,a+1+n);  /*使用标准模板库STL中的的sort函数排序，默认由小到大。a是数组名表示数组的首地址。
    如由大到小排序，则sort(a+1,a+1+n,comp)。其中comp为自定义函数，参考《C++实用技巧与模板库--排序算法》。
	int comp(const int &a,const int&b) {return a>b;}
	也可以bool comp(const int &a,const int&b) {return a>b;} 
	*/  

for(i=1;i<=r;i++)  //前r个人为一组，每个人的打水时间即为自己的装水时间。 
	b[i]=a[i];
for(i=r+1;i<=n;i++)  //从第r+1个人开始，每个人的打水时间为等待时间+装水时间。 
	b[i]=b[i-r]+a[i];
for(i=1;i<=n;i++)    //累加每个人的打水时间
	sum=sum+b[i];   
	
printf("%d\n",sum);   //输出结果 
return 0;

}