水龙头接水问题

题目描述

学校里有一个水房，水房里一共装有 m 个龙头可供同学们打开水，每个龙头每秒钟的供水量相等，均为 11。

现在有 n 名同学准备接水，他们的初始接水顺序已经确定。将这些同学按接水顺序从 11 到 n 编号，i 号同学的接水量为 wi​。接水开始时，11 到 m 号同学各占一个水龙头，并同时打开水龙头接水。当其中某名同学 j 完成其接水量要求 wj​ 后，下一名排队等候接水的同学 k 马上接替 j 同学的位置开始接水。这个换人的过程是瞬间完成的，且没有任何水的浪费。即 j 同学第 x 秒结束时完成接水，则 k 同学第 x+1 秒立刻开始接水。若当前接水人数 ′n′ 不足 m，则只有 ′n′ 个龙头供水，其它 ′m−n′ 个龙头关闭。

现在给出 n 名同学的接水量，按照上述接水规则，问所有同学都接完水需要多少秒。

输入格式

第一行两个整数 n 和 m，用一个空格隔开，分别表示接水人数和龙头个数。

第二行 n 个整数 1,2,…,w1​,w2​,…,wn​，每两个整数之间用一个空格隔开，wi​ 表示 i 号同学的接水量。

输出格式

一个整数，表示接水所需的总时间。

输入输出样例

输入 #1

5 3
4 4 1 2 1

输出 #1

4

输入 #2

8 4
23 71 87 32 70 93 80 76

输出 #2

163

说明/提示

【输入输出样例 #1 说明】

第 1 秒，3 人接水。第 1 秒结束时，1,2,3 号同学每人的已接水量为 1,3 号同学接完水，4 号同学接替 3 号同学开始接水。

第 2 秒，3 人接水。第 2 秒结束时，1,2 号同学每人的已接水量为 2,4 号同学的已接水量为 1。

第 3 秒，3 人接水。第 3 秒结束时，1,2 号同学每人的已接水量为 3,4 号同学的已接水量为 2。4 号同学接完水，5 号同学接替 4 号同学开始接水。

第 4 秒，3 人接水。第 4 秒结束时，1,2 号同学每人的已接水量为 4,5 号同学的已接水量为 1。1,2,5号同学接完水，即所有人完成接水的总接水时间为 4 秒。

【数据范围】

1≤n≤10000，1≤m≤100，m≤n；

1≤wi​≤100。

看到这题首先想到的是模拟，可以想到的是一个水龙头接完水了，后面的人补到那个人上，用循环，接水的人-1，时间+1。或者使用优先队列。但是还有一种特殊的方法就是我们可以直接用人去接水龙头里的水，接完后面的人补上，计算三个水龙头的值就行了，简便很多。

首先我们将前三个接水量赋值给三个水龙头

for(int i=1; i<=m; i++) {
		a[i] = w[i];
	}

然后将后面的接水量加入到这三个水龙头中去，先找出三个水龙头的最小值，因为它最先接完，后面的人补到这个位置，所以将后一个人的接水量加到最小的水龙头中，循环完成此方案。

for(int i=m+1; i<=n; i++) {
		int j = 1;
		//找出前三个水龙头中最小的
		for(int k=2; k<=m; k++) {
			if(a[j] > a[k]) {
				j = k;
			}
		}
		a[j] += w[i];
	}

最后找出这三个水龙头的最大值就是所用的时间

int max = 0;
	for(int i=1; i<=m; i++) {
		if(a[i] > max) {
			max = a[i];
		}
	}

是不是觉得恍然大悟

完整代码如下

#include <stdio.h>

int main()
{
	int n,m,a[10005],w[10005];
	scanf("%d %d",&n,&m);
	for(int i=1; i<=n; i++) {
		scanf("%d",&w[i]);
	}
	for(int i=1; i<=m; i++) {
		a[i] = w[i];
	}
	for(int i=m+1; i<=n; i++) {
		int j = 1;
		//找出前三个水龙头中最小的
		for(int k=2; k<=m; k++) {
			if(a[j] > a[k]) {
				j = k;
			}
		}
		a[j] += w[i];
	}
	int max = 0;
	for(int i=1; i<=m; i++) {
		if(a[i] > max) {
			max = a[i];
		}
	}
	printf("%d",max);
	return 0;
}