6.1 接水问题（模拟，优先队列）

来源：NOIP2010普及组 https://ac.nowcoder.com/acm/contest/238/C

学校里有一个水房，水房里一共装有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个整数，表示接水所需的总时间。
示例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。

题解一：模拟，若有一个水龙头为零，立马让后面的补上，退出条件是人已经都上场了。然后再判断当前m个水龙头的最大值，即需要的时间数。用循环次数加上最大值就是最后答案。。。

#include <cstdio>
#include <queue>
#include <stack>
#include <cstdio>
#include <string>
#include <iostream>
#include <algorithm>
using namespace std;

const int N = 10010;
int water[N];

int main()
{
    int n, m;//5 3
    cin>>n>>m;
    for(int i = 0; i < n; i++)
        cin>>water[i];

    int j= m;
    int res = 0;
    //到达最后一个水龙头退出
    while(j < n)
    {
        //一次循环完m个水龙头，另i从0 继续下一次循环
        for(int i = 0; i < m; i++)
        {
            if(water[i] > 0) water[i] -= 1;
            //一个水龙头结束，后面的人顶替
            if(water[i] == 0) water[i] = water[j++];//3
        }
        //m个水龙头都减1后，时间为 1 ，累加
        res++;
    }

    int maxn = 0;
    //当前还需要的时间是m个水管的最大值
    for(int i = 0; i < m; i++)
        maxn = max(maxn, water[i]);
    //输出
    cout<<res + maxn<<endl;

    return 0;
}

题解二：直接用priority_queue来做，从小到大排序。greater < int > ,每次取出队首，即当前最需要换人的水龙头，将下一个人时间和当前水龙头累积时间起来，继续入队。最后找到队尾，即用时最长的，就是答案！

注意：一定要初始化每个水龙头的值为0，队列存储的是每个水龙头的累计时间。

#include <queue>
#include <stack>
#include <queue>
#include <cstdio>
#include <string>
#include <iostream>
#include <algorithm>
#define mod 20123
using namespace std;
typedef long long LL;

const int N = 10010;

int arr[N];

int main()
{
    int n, m;

    cin >> n >> m;
    for(int i = 0; i < n; i++)
        cin >> arr[i];
    priority_queue<int, vector<int>, greater<int>> heap;

    //m个水龙头起始都为零
    for(int i = 0; i < m; i++) heap.push(0);

    for(int i = 0; i < n; i++)
    {
        //取出最小，即最早需要换人的水龙头
        int t = heap.top();
        //删掉
        heap.pop();
        //将当前人与下一个接水的人总时间入队
        heap.push(t + arr[i]);
    }

    int res;
    //答案就是队尾时间最大的那个
    while(!heap.empty())
    {
        res = heap.top();
        heap.pop();
    }
    cout<< res <<endl;
    return 0;
}