蓝桥杯算法训练：接水问题java

试题 算法训练 接水问题

学校里有一个水房，水房里一共装有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 个整数，表示接水所需的总时间。
样例输入
5 3
4 4 1 2 1
样例输出
4
样例输入
8 4
23 71 87 32 70 93 80 76

import java.util.Arrays;
import java.util.Scanner;

public class jieshuiwenti {
    public static void main(String[] args) {
        Scanner s=new Scanner(System.in);
        int ans=0;
        int n=s.nextInt();//人
        int m=s.nextInt();//水龙头
        int[] arr=new int[n];//任务量
        int[] arrshui=new int[m];//每个水龙头当前的工作量
        for (int i=0;i<n;i++){
            arr[i]=s.nextInt();
        }
        if (m>=n){
            Arrays.sort(arr);
            System.out.println(arr[arr.length-1]);
        }else {
            for (int i=0;i<m;i++){
                arrshui[i]=arr[i];
            }
            Arrays.sort(arrshui);
            int i=m;
            while (i!=n){
                int temp=arrshui[0];
                for (int j=0;j<m;j++){
                    arrshui[j]-=temp;
                }
                ans+=temp;
                arrshui[0]=arr[i];
                Arrays.sort(arrshui);
                i++;
            }
        }
        Arrays.sort(arrshui);
        ans+=arrshui[arrshui.length-1];
        System.out.println(ans);
    }
}