算法提高 排队打水问题

 

问题描述

　　有n个人排队到r个水龙头去打水，他们装满水桶的时间t1、t2………..tn为整数且各不相等，应如何安排他们的打水顺序才能使他们总共花费的时间最少？

输入格式

　　第一行n，r (n<=500,r<=75)
　　第二行为n个人打水所用的时间Ti (Ti<=100)；

输出格式

　　最少的花费时间

样例输入

3 2
1 2 3

样例输出

7 

数据规模和约定

　　其中80%的数据保证n<=10

/*# include <stdio.h>
int main()
{
    int n, t, cnt, i, j;

    scanf("%d", &t);

    for(j=1; j<=t; j++){
        scanf("%d", &n);
        cnt = 0;

        for(i=2; n!=1; i++){
            while(n%i == 0){
                n /= i;
                cnt++;
            }
        }
        if(cnt == 2){
            printf("Yes\n");
        }
        else{
            printf("No\n");
        }
    }
    return 0;
}
*/

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int main() {
    int n, r, sum=0;

    scanf("%d %d", &n, &r);

    int *a = new int[n];

    for(int i = 0; i < n; i++)
    {
        scanf("%d", &a[i]);
        sum += a[i];
    }


    sort(a, a+n);

    int cnt = n/r;
    int ct = n%r;
int t= r;
for(int j=0; j<r; j++)
    {
        int tmp = cnt;
        if(ct)
        {

            for(int i=0; i<cnt+1; i++)
            {
                sum += a[i*r+j]*(tmp--);


            }
            ct--;
        }
    else{    tmp = cnt-1;
        for(int i=0; i<cnt; i++)
        {
            sum += a[i*r+j]*(tmp--);
 //           cout << sum << endl;

        }}
//        cout <<endl << sum << endl;

        //
//        cnt--;

    }
    cout << sum << endl;

    delete [] a;

    return 0;
}