接水问题-蓝桥杯

题目描述

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

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

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

输入描述

第 1 行 2 个整数 nn 和 mm用一个空格隔开，分别表示接水人数和龙头个数。 第 2 行 n 个整数 w_1、w_2、\cdots、w_nw1​、w2​、⋯、wn​，每两个整数之间用一个空格隔开，w_iwi​ 表示 ii 号同学的接水量。

输出描述

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

输入输出样例

示例 1

输入

5 3 
4 4 1 2 1 

输出

4

示例 2

输入

8 4 
23 71 87 32 70 93 80 76 

输出

163

运行限制

• 最大运行时间：1s
• 最大运行内存: 128M

代码

import java.util.Scanner;
// 1:无需package
// 2: 类名必须Main, 不可修改

public class Main {
    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        //在此输入您的代码...
        int n = scan.nextInt();
        int m = scan.nextInt();
        int[] arr = new int[n];
        for (int i = 0; i < n; i++) {
            arr[i] = scan.nextInt();
        }
        int max = arr[0];
        if(m>n){
            for (int i = 1; i < n; i++) {
                max = arr[i]>max?arr[i]:max;
            }
            System.out.println(max);
            return;
        }
        for (int i = m; i < n; i++) {
            int index=0;
            for (int j = 1; j < m; j++) {
                index = arr[j]<arr[index]?j:index;
            }
            arr[index] += arr[i];
        }
        for (int i = 1; i < n; i++) {
            max = arr[i]>max?arr[i]:max;
        }
        System.out.println(max);
    }
}