java线程打水问题_Java实现 蓝桥杯VIP 算法提高 打水问题

算法提高 打水问题

时间限制：1.0s 内存限制：512.0MB

问题描述

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.io.BufferedReader;

import java.io.InputStreamReader;

import java.util.Arrays;

import java.util.StringTokenizer;

public class 打水问题 {

public static void main(String args[])throws Exception {

BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

StringTokenizer st = new StringTokenizer(br.readLine());

int n = Integer.parseInt(st.nextToken()), m = Integer.parseInt(st.nextToken());

st = new StringTokenizer(br.readLine());

int[] time = new int[n];

for(int i=0; i

Arrays.sort(time);

int total = 0;

for(int i=1; i<=m; i++){

int p = n/m;

if(n%m>=i)p++;

for(int index = i-1; index

total += time[index]*(p-1);

}

System.out.println(total);

}

}