(蓝桥杯)试题 算法训练 排队接水2

Posted 2021-09-25 nuist__NJUPT

试题 算法训练 排队接水2

资源限制
时间限制：1.0s 内存限制：256.0MB
问题描述
　　有N个人排队到M个水龙头去打水，他们装满水桶的时间T1,T2……Tn为整数且各不相等，应如何安排他们的打水顺序才能使他们花费的总时间最少？
输入格式
　　第1行：两个整数n和m，n表示人的个数，m表示水龙头的个数；
　　第2行，n个数分别表示n个人装水的时间
　　数据范围：m<=n/3,n<=1000,t<3000。
输出格式
　　一个整数，表示总花费的最少时间。
样例输入
6 2
5 4 6 2 1 7
样例输出
40

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

public class Main {
    public static void main(String[] args){
        Scanner input = new Scanner(System.in) ;
        int n = input.nextInt() ; //接水的人数
        int m = input.nextInt() ; //水龙头个数
        int [] times = new int [n] ; //n个人装水的时间
        int [] wait = new int [m] ; //m个水龙头等待时间
        for(int i=0; i<n; i++){
            times[i] = input.nextInt() ;
        }
        Arrays.sort(times); //装水时间由小到大排序
        int sum = 0 ;
        for(int i=0; i<n;){
            for(int j=0; j<m && i<n; j++,i++){
                wait[j] += times[i] ;
                sum += wait[j] ;
            }
        }
        System.out.println(sum) ;
    }
}