接水问题

/*题目描述

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

输入

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

输出

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

样例输入

5 3
4 4 1 2 1
样例输出

4
提示

1 ≤ n ≤ 10000，1 ≤ m ≤ 100 且 m ≤ n；

1 ≤ w i ≤ 100。



每次将用时最少的同学取出，总时间加上用掉的时间，并且记得把正在接水的同学的时间也要减去。。。

*/

package test;

import java.util.PriorityQueue;
import java.util.Queue;
import java.util.Scanner;

public class 接水问题{
    static int Max=(int) (1e4+10);
    static int INF = 0x3f3f3f3f;
    static int n,m;
    static int[] w;
    static Queue<Integer> queue=new PriorityQueue<>();//优先级队列
    public static void main(String agr[]){
        Scanner input=new Scanner(System.in);
        n=input.nextInt();
        m=input.nextInt();
        int res = INF;
        int ses=-1;
        w=new int[n+1];
        for (int i = 0; i < n; i++)
        {
            w[i]=input.nextInt();
            res =Math.min(res, w[i]);//最少接水量
            ses=Math.max(ses, w[i]);//最多接水量
        }

        if (n <= m)//如果水龙头的数量比接水的人多
        {
            System.out.println(ses);
            return ;
        }

        for (int i = 0; i < m; i++)//先初始化水龙头的开始等水量
        {
            queue.add(w[i]);
        }

        for (int i = m; i < n; i++)//找出用时最小的加上后一个同学的用时形成新的队列
        {
            int t = w[i] + queue.peek();//优先级队列每次取出最小元素
            queue.poll();//将原有的最小元素出队列后再加入新生成的元素
            queue.add(t);
        }

        res = 0;
        while (!queue.isEmpty())
        {
            res = Math.max(res, queue.peek());
            queue.poll();
        }

        System.out.println(res);//所有同学接完水后找出最大值，则为接水总时长
    }
}