题目 1629: 蓝桥杯算法训练VIP-接水问题

题目描述:

学校里有一个水房，水房里一共装有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  说明
第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秒。

代码:

package lanqiao;

import java.math.BigInteger;
import java.text.DecimalFormat;
import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt(); //接水人数
        int m = sc.nextInt(); //水龙头人数

        int[] arr1 = new int[n];
        int[] arr2 = new int[m];
        for(int i = 0;i < n;i ++)
        {
            arr1[i] = sc.nextInt();
        }

        if(n <= m)
        {
            int max = 0;
            for(int i = 0;i < n; i++)
            {
                max = max < arr1[i] ? arr1[i]:max;
            }
            System.out.println(max);
        }
        else{
            for(int i = 0;i < m;i ++)
            {
                arr2[i] = arr1[i];
            }
            int index = m;
            for(int i = m;i < n;i ++)
            {
                int min = arr2[0];
                for(int j = 0;j < m;j ++)
                {
                    min = min > arr2[j]?arr2[j]:min;
                }
                for(int j = 0;j < m;j ++)
                {
                    if(arr2[j] == min)
                    {
                        arr2[j] += arr1[index];
                        index ++;
                    }
                    if(index == n)
                    {
                        break;
                    }
                }
            }

            int max = 0;
            for(int i = 0;i < m;i++)
            {
                max = max < arr2[i]?arr2[i]:max;
            }

            System.out.println(max);
        }

    }
}