NOIP - 接水问题

题目描述

学校里有一个水房，水房里一共装有 m 个龙头可供同学们打开水，每个龙头每秒钟的
供水量相等，均为 1。
现在有 n 名同学准备接水，他们的初始接水顺序已经确定。将这些同学按接水顺序从 1
到 n 编号，i 号同学的接水量为 w i 。接水开始时，1 到 m 号同学各占一个水龙头，并同时打
开水龙头接水。当其中某名同学 j 完成其接水量要求 w j 后，下一名排队等候接水的同学 k
马上接替 j 同学的位置开始接水。这个换人的过程是瞬间完成的，且没有任何水的浪费。即
j 同学第 x 秒结束时完成接水，则 k 同学第 x+1 秒立刻开始接水。若当前接水人数 n’不足 m，
则只有 n’个龙头供水，其它 m−n’个龙头关闭。
现在给出 n 名同学的接水量，按照上述接水规则，问所有同学都接完水需要多少秒。

输入

输入第 1 行 2 个整数 n 和 m，用一个空格隔开，分别表示接水人数和龙头个数。
第 2 行 n 个整数 w 1 、w 2 、……、w n ，每两个整数之间用一个空格隔开，w i 表示 i 号同
学的接水量。

输出

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

样例输入

5 3
4 4 1 2 1

样例输出

4

提示

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

1 ≤ w i ≤ 100。

每次将用时最少的同学取出，总时间加上用掉的时间，并且记得把正在接水的同学的时间也要减去。。。

代码中有注释

#include<cstdio>
#include<iostream>
using namespace std;

int w[10010];  //每个同学的接水量
int lastw[10010]; //每个水龙头位置的同学还需要的接水量

int main()
{
    int n, m;
    int time = 0;

    scanf("%d%d",&n,&m);
    for (int i=1; i<=n; i++) scanf("%d",&w[i]);

    for (int i=1; i<=m; i++) lastw[i] = w[i];  //一开始的时候，前m个同学会同时去接水
    int i = m + 1;
    while (i <= n)
    {
        int time_min = (1<<31)-1;
        for (int j=1; j<=m; j++)  //在水龙头前的m个同学中，找到最先接完水的同学所用的时间
        {
            time_min = min(time_min,lastw[j]);
        }
        time += time_min;  //把最先接完水的同学所用的时间累加到总时间中
        for (int j=1; j<=m; j++)  //计算最快的同学接完水的时候，在水龙头前的m个同学还需要的水量
        {
            lastw[j] -= time_min;
            if (lastw[j] == 0)
            {
                lastw[j] = w[i];
                i++;
            }
        }
    }
    int time_min = 0;
    for (int i=1; i<=m; i++)  //这时只剩下最后m个同学了，要让所有的同学都接完水，要找最后一个同学接完水所用的时间
    {
        time_min = max(time_min,lastw[i]);
    }
    time += time_min;

    printf("%d\n",time);

    return 0;
}