蓝桥杯算法提高VIP-排队打水问题-题解(Java代码)

import java.util.Scanner;
import java.util.Arrays;
public class Main {
    public static void main(String[] args) {
        Scanner scanner=new Scanner(System.in);
        int n=scanner.nextInt();
        int r=scanner.nextInt();
        int []ns=new int[n];
        for (int i = 0; i < n; i++) {
            ns[i]=scanner.nextInt();
        }
        Arrays.sort(ns);
        int sum=0;
        int []nv=new int[r];
        int k=0;
        int v=0;
        for (int i = 0; i < n; i++) {
                nv[k]+=ns[i];
                sum+=nv[k];
            k++;
            if (k>=r) {
                k=0;
            }
        }
        System.out.println(sum);
    }
}