信息学奥赛一本通：1233：接水问题

【题目描述】

学校里有一个水房，水房里一共装有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

【提示】

样例输入#2：

8 4

23 71 87 32 70 93 80 76

样例输出#2：

163

//在原数组上做，以数组前部分为水龙头 
//每个水龙头减1，判断为0吗，如是m后的一人来顶上 
//直到无人能来，排序，再加上减的数就是了 
//m大于n时，就直接排序，找最大输出可 
//Writed by Wangzhimin 2023/11/21 
//数组开得不够大时1005，还有运行错误呢 
#include<bits/stdc++.h>
using namespace std;
int a[100005];
int main()
{
    int n,m,i,sum,num=0;
    scanf("%d%d",&n,&m);
    for(i=0;i<n;i++)
    {
        scanf("%d",&a[i]);
    }
    a[n]=0;//防备 
    sum=m;//为n大时用的 
    if(m>=n)
    {
        sort(a,a+n);
        printf("%d",a[n-1]);
    }
    else
    {
        while(1){
            if(a[sum]==0)
            break;
            for(i=0;i<m;i++)
            a[i]--;//每个水龙头减1 
            num++;//计数减了几次 
            for(i=0;i<m;i++)
            {
                if(a[i]==0)//减没了，就换人 
                {
                    a[i]=a[sum];
                    sum++;
                }
               }
        }
        sort(a,a+m);
        printf("%d",num+a[m-1]);        
    }
 }