CSP-J复赛冲刺必刷题 | P1190 接水问题

学习C++从娃娃抓起！记录下CSP-J备考学习过程中的题目，记录每一个瞬间。

附上汇总贴：CSP-J复赛冲刺必刷题 | 汇总_热爱编程的通信人的博客-CSDN博客

---

【题目描述】

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

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

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

【输入】

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

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

【输出】

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

【代码详解】

#include <bits/stdc++.h>
using namespace std;

int main()
{
    int n, m;
    cin >> n >> m;
    int w[n+5];
    for (int i=1; i<=n; i++) cin >> w[i];
    int lt[m+5];
    for (int j=1; j<=m; j++) lt[j] = j;
    long long time = 0;
    int wait = m+1;
    while (true) {
        time++;
        for (int j=1; j<=m; j++) {
            if (lt[j]<0) continue;
            w[lt[j]]--;
            if (w[lt[j]]==0) {
                lt[j] = wait;
                wait++;
                if (wait>n+1) lt[j]=-1;
            }
        }
        if (wait<=n) continue;
        bool finish=true;
        for (int i=1; i<=n; i++) {
            if (w[i]>0) {
                finish = false;
                break;
            }
        }
        if (finish) break;
    }
    cout << time;
    return 0;
}

【运行结果】

5 3
4 4 1 2 1
4