描述 学校里有一个水房,水房里一共装有 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 用例输出 1 4 用例输入 2 8 4 23 71 87 32 70 93 80 76 用例输出 2 163c++代码
时间: 2024-03-22 17:37:30 浏览: 99
```cpp
#include <iostream>
#include <queue>
using namespace std;
int main() {
int n, m;
cin >> n >> m;
priority_queue<pair<int, int> > q; // 用一个大根堆存储剩余水量和同学序号
for (int i = 1; i <= n; i++) {
int w;
cin >> w;
q.push(make_pair(w, i));
}
int ans = 0;
vector<int> remain(m, 0); // 存储龙头剩余水量
while (!q.empty()) {
auto top = q.top();
q.pop();
int w = top.first, id = top.second;
int min_remain = remain[0];
int min_index = 0;
for (int i = 1; i < m; i++) { // 找到剩余水量最小的龙头
if (remain[i] < min_remain) {
min_remain = remain[i];
min_index = i;
}
}
if (min_remain == 0) { // 如果所有龙头都用完了水,等到最早的龙头用完
ans += remain[min_index];
for (int i = 0; i < m; i++) {
remain[i] -= remain[min_index];
}
}
int t = min(w, m - min_remain); // 计算当前同学接水所需时间
ans += t;
remain[min_index] += t; // 更新龙头剩余水量
if (w > t) { // 如果当前同学没有接完水,继续加入堆中
q.push(make_pair(w - t, id));
}
}
cout << ans << endl;
return 0;
}
```
阅读全文