codevs 3377 [Mz]接水问题2

题目描述 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名同学的接水量，按照上述接水规则，问所有同学都接完水需要多少秒。

特别地，同学们在打水前排好了队，接水所用时间更长的先接。

（ps：出题人本来想设计一个贪心的题目，然后发现用贪心做有反例，只能强改题目，在此声明道歉。不要在意样例解释。）

输入描述 Input Description

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

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

输出描述 Output Description

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

样例输入 Sample Input

5 3

4 4 1 2 1

样例输出 Sample Output

4

数据范围及提示 Data Size & Hint

【输入输出解释】

第 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 号同学接完水，即所有人完成接水。 

【数据范围】
对于 30%的数据，n≤10,000，m≤1,000；
对于全部的数据，1≤m≤n≤1,000,000，1≤m≤100,000。

思路：

先从小到大排序，然后先将前m个数放入小根堆；

然后取出堆顶元素，加a[i]后在放入堆中；

输出堆中最大元素。

代码：

#include<cstdio>
#include<queue>
#include<algorithm>
using namespace std;
priority_queue<int,vector<int>,greater<int> >a;
int n,m,b[1000001],maxn;
bool cmp(int x,int y)
{
    return x>y;
}
int main()
{
    int i,j;
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;i++)
    {
        scanf("%d",&b[i]);
        if(n<=m)
          maxn=max(maxn,b[i]);
    }  
    if(n<=m)
    {
        printf("%d",maxn);
        return 0;
    }
    sort(b+1,b+n+1,cmp);
    for(i=1;i<=m;i++)
      a.push(b[i]);
    for(i=m+1;i<=n;i++)
    {
        int x=a.top();
        a.pop();
        x+=b[i];
        a.push(x);
    }
    while(a.size()>1)
      a.pop();
    printf("%d",a.top());
    return 0;
}