蓝桥杯算法提高VIP-打水问题（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>

int faucet(int n, int m, int* t) {
	int i = 0, j = 0;
	int a[1010] = { 0 };//各个水龙头的等待时间，初始为0
	int waitTime=0;//等待时间
	while (i < n) {
		
		for (j = 0;j < m;j++) {
			waitTime += a[j];//记录等待时间
			a[j] += t[i];//打水
			i++;//下一个人
			if (i >= n)
				break;
		}

	}
	return waitTime;
}
int main() {
	int n, m;
	int t[1010];
	scanf("%d %d", &n, &m);
	int i, j;
	for (i = 0;i < n;i++) {
		scanf("%d", &t[i]);
	}
	for (i = 0;i < n;i++) {//打水时间从小到大排序
		for (j = 0;j < i;j++) {
			if (t[j] > t[i]) {
				int k = t[i];
				t[i] = t[j];
				t[j] = k;
			}
		}
	}
	
	printf("%d", faucet(n, m, t));

	return 0;
}