算法训练 接水问题(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
分析
注意看题目,题目要求的是所有人的等待时间之和最短 ,既然如此,我们就应该思考什么情况下时间之和最短,如果先让打水时间最短的人去打水,那么后面的人等待时间自然就是最短的.
参考代码
#include<stdio.h>
#include<stdlib.h>
int cmp (const void *a,const void *b){//qsort要求的比较函数
return *(int*)a-*(int*)b;
}
int main (void){
int i,j,n,m,time=0;
scanf("%d%d",&n,&m);
int p[1005]={0},q[1005]={0};//p为打水的人,q为等待的时间
for(i=0;i<n;i++){
scanf("%d",p+i);
}
qsort(p,n,sizeof(p[0]),cmp);//快速排序
for(i=0;i<n;i++){
qsort(q,m,sizeof(q[0]),cmp);//从小到大排序等待时间
time+=q[0];//这里有点难以理解,如果不明白,可自行按循环推导
q[0]+=p[i];
}
printf("%d",time);
return 0;
}
如有不正之处,敬请指正!