【10NOIP普及组】接水问题

http://ybt.ssoier.cn:8088/problem_show.php?pid=1950
【题目描述】
学校里有一个水房，水房里一共装有m 个龙头可供同学们打开水，每个龙头每秒钟的供水量相等，均为1。

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

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

【输入】
第1 行2 个整数n 和m，用一个空格隔开，分别表示接水人数和龙头个数。

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

【输出】
只有一行，1 个整数，表示接水所需的总时间。

【输入样例】
5 3
4 4 1 2 1
【输出样例】
4
【提示】
【输入输出样例说明】

第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 秒。

【输入输出样例2】

输入:

8 4
23 71 87 32 70 93 80 76
输出:

163
【数据范围】

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

1≤wi≤100。

#include<bits/stdc++.h>
using namespace std;
#define N 10005
int a[N];
long long ans;
int main(){
	int n, m, tmp = 105, j=2;
	cin >> n >> m;
	for(int i=1; i<=n; i++)
		cin >> a[i];
	m = min(n,m);
	for(int i=m; i<=n; ){
		sort(a+1, a+m+1);
		ans += a[1];
		int tmp = a[1], add = 0;
		for(int j=1; j<=m; j++){
			a[j] -= tmp;
			if(a[j] == 0)//计算每次打完水的人数
				++add;
		}
		int t = 0, r = i+add;
		while(i < r && i <= n){//从后面补相应的add人数上来，注意i<r!不能等于！
			a[++t] = a[++i];
		}	
	}
	sort(a+1, a+m+1);
	ans += a[m];
	cout << ans;
	return 0;
}