【贪心】接水问题

【贪心】接水问题

题面

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

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

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

思路

一共m个水龙头，遍历n个人，每次把人安排在接水时间最少的一个水龙头。
使用：优先队列

代码

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

const int N = 1e4 + 10;

priority_queue<int, vector<int>,greater<int>> pq;
//小根序优先队列

int main() {
    int n, m;
    cin >> n >> m;
    for(int i = 1; i <= m; i++) {
        int t = 0;
        pq.push(t);
        //输入初始数据,每个水龙头均为0
    }
    for(int i = 1; i <= n; i++) {
        int t;
        cin >> t;
        t += pq.top();//加在时间最少的水龙头上
        pq.pop();
        pq.push(t);
    }
    while(pq.size() > 1) {
        pq.pop();
        //只留下最大的那个水龙头
    }
    cout<<pq.top();
	return 0;
}