蓝桥杯练习----算法提高----ADV-104 打水问题（Java解法）

蓝桥杯练习----算法提高----ADV-104 打水问题（Java解法）

问题描述

N个人要打水，有M个水龙头，第i个人打水所需时间为Ti，请安排一个合理的方案使得所有人的等待时间之和尽量小。

输入格式

第一行两个正整数N M 接下来一行N个正整数Ti。N,M<=1000，Ti<=1000

输出格式

最小的等待时间之和。（不需要输出具体的安排方案）
　　
样例输入
7 3
3 6 1 4 2 5 7
样例输出
11

思路

将N个人按照Ti从小到大的顺序依次分配到M个龙头打水。
Ti从小到大排序为1，2，3，4，5，6，7，将他们依次分配到3个龙头，则去龙头一打水的为1，4，7；去龙头二打水的为2,5；去第三个龙头打水的为3,6。
　　第一个龙头打水的人总等待时间 = 0 + 1 + (1 + 4) = 6
　　第二个龙头打水的人总等待时间 = 0 + 2 = 2
　　第三个龙头打水的人总等待时间 = 0 + 3 = 3
　　所以总的等待时间 = 6 + 2 + 3 = 11

代码如下

import java.util.Arrays;
import java.util.Scanner;
public class Main {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n =sc.nextInt();
		int m =sc.nextInt();
		int [] arr =new int [n];
		for (int i = 0; i < arr.length; i++) {
			arr[i]=sc.nextInt();
		}
		Arrays.sort(arr);
		int sum=0;
		for (int i = 0; i < m; i++) {
			int temp=arr[i];
			for (int j = i+m; j < arr.length; j+=m) {
				sum+=(temp);
				temp+=arr[j];
			}
		}
		System.out.print(sum);
		}
}

参考代码 借鉴了解题思路 原文链接：https://blog.csdn.net/mcshangyun/article/details/88086004

只是简化了下代码，感觉网上好多大佬写的都看不懂 有的还没学，
这样看着简单易懂 。