算法训练接水问题

Description

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

Input

输入描述:

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

输入样例:

5 3

4 4 1 2 1

Output

输出描述:

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

输出样例:

Sample Input 1

参考上文

Sample Output 1

参考上文

Hint

HINT:时间限制：1.0s 内存限制：64.0MB

　　第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 秒。

思路：典型的排队问题，队伍最长的就是所有同学都打到水的最短时间，堆的灵活运用

#include <iostream>
#include <cstring>
#include <algorithm>
#include <queue>

using namespace std ;


int main(){
    int n, m ;
    cin >> n >> m ;
    priority_queue<int,vector<int>,greater<int>> pq ;
    for(int i=1;i<=n;i++){
        int a ;
        cin >> a ;
        if(i<=m){
            pq.push(a) ;
        }else{
            int tmp = pq.top() ;
            pq.pop() ;
            pq.push(tmp+a) ;
        }
    }
    int ans = 0 ;
    while(pq.size()>1){
        pq.pop() ;
    }
    ans = pq.top() ;
    cout << ans << endl ;
    
    return 0 ;
}

...

算法训练 接水问题

算法训练接水问题