蓝桥杯算法训练 接水问题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
样例输出
163
输入输出样例 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 秒。
数据规模和约定
　　1 ≤ n ≤ 10000，1 ≤m≤ 100 且m≤ n；
　　1 ≤ wi ≤ 100。

思路：
网上搜了一下，发现没有让我满意的代码（虽然我自己写的也不是特别好）,于是就自己写了一个，思路都在注释,代码如下：

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

public class Main {

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner read = new Scanner(System.in);
		int n = read.nextInt();//人数
		int m = read.nextInt();//水龙头个数
		int[] water = new int[n];
		int[] num = new int[m];//正在接水的，还要接多久
		for(int i=0;i<n;i++)
		{
			water[i] = read.nextInt();
		}
		//开始接水，两种情况，第一种人数《=水龙头，或者大于
		if(n<=m)
		{
			Arrays.sort(water);
			System.out.println(water[n-1]);
		}
		else
		{
			//因为我们求的是接水总时间,所以我们将最久的水龙头定在num[n-1]

			int human = 0;//几个人接完水了（包括正在接水的）
			for(int i=0;i<num.length;i++)
			{
				num[i] = water[i];
				human++;
			}
			Arrays.sort(num);
			int count = 0;//计算时间
			while(num[m-1]>0)
			{
				Arrays.sort(num);
				int mark = 0;//记录最小的接水时间
				for(int i=0;i<num.length;i++)//找到最小的接水时间
				{
					if(num[i]!=0)
					{
						mark = num[i];
						count +=mark;
						break;
					}
				}
				
				for(int i=0;i<num.length;i++)//减去最小的接水时间
				{
					if(num[i]!=0)
					{
						num[i] -= mark;
					}
				}
				//肯定有人接完水了,有人的话就补人
				int countWater = 0;//记录有多少人接完水了
				int count2 = 0;//num肯定是从0开始的人接完水,一共有多少个人接完？？countWater个
				if(human<water.length)
				{
					for(int i=0;i<num.length;i++)
					{
						if(num[i]==0)
							countWater++;
					}
					//剩下的人够不够补充
					if(human+countWater>water.length)//不够
					{
						for(int i=human;i<water.length;i++)
						{
							num[count2] = water[i];
							count2++;
						}
					}
					else
					{
						for(int i=human;i<(human+countWater);i++)
						{
							num[count2] = water[i];
							count2++;
						}
					}
				}
				human += countWater;
				
			}
			System.out.println(count);
		}
		read.close();
	}

}