打水问题

算法提高 打水问题

 时间限制: 1 Sec  内存限制: 512 MB

题目描述

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

来源

思路： 可以将需要打水人所用时间升序排列，按时间从小到大打水。

#include<stdio.h>
#include<iostream>
#include<algorithm>
using namespace std;
int main()
{
    int n,m,t[1005],i,j,s=0,s1;
    cin>>n>>m;
    for(i=0;i<n;i++)
        cin>>t[i];
    sort(t,t+n);
    for(j=0;j<m;j++)
    {
        i=j;
        s1=0;
        while(i<n-m)
        {
           s1+=t[i];
           s+=s1;
           i+=m;
        }
    }
    cout<<s<<endl;
    return 0;
}