接水问题

题目描述

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

思路：

由于接水的顺序已经确定了，所以我们要做的只是模拟这个过程——m个龙头，一旦有人接满水，就让下一个人去那个位置，并将时间更新为还需要的时间。

代码：

package AHuawei;

import java.util.Scanner;

public class Main {
	public static void main(String[] args) {
		Scanner scan = new Scanner(System.in);
		while(scan.hasNext()) {
			int n = scan.nextInt(); //n人接水
			int m = scan.nextInt(); //m个龙头
			int[] data = new int[n]; //每人的接水量
			int[] needTime = new int[n];
			int max = 0;
			for (int i = 0; i < n; i++) {
				data[i] = scan.nextInt();
				needTime[i] = data[i];
				max = Math.max(data[i], max);
			}
			int min = 0;
			int minPos = 0;
			if(m == n) {
				System.out.println(max);
				continue;
			}
			for (int i = m; i < n; i++) {
				min = Integer.MAX_VALUE;
				minPos = -1;
				for (int j = 0; j < m; j++) {
					if(min > needTime[j]) {
						min = needTime[j];
						minPos = j;
					}
				}
				needTime[minPos] += data[i];
				max = Math.max(needTime[minPos], max);
			}
			System.out.println(max);
		}
		scan.close();
	}
}