1022: 算法提高 打水问题
时间限制: 1 Sec 内存限制: 512 MB提交: 10 解决: 4
[ 提交][ 状态][ 讨论版]
题目描述
N个人要打水,有M个水龙头,第i个人打水所需时间为Ti,请安排一个合理的方案使得所有人的等待时间之和尽量小。
输入
第一行两个正整数N M 接下来一行N个正整数Ti。
N,M<=1000,Ti<=1000
输出
最小的等待时间之和。(不需要输出具体的安排方案)
样例输入
7 33 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
例如:1 2 3
3 4 5 1 2 3
6 7 8 1 2 3 34 5
9 10 11 1 2 3 3 4 5 6 7 8
12 13 1 3 6 9
2 4 7 10
#include<iostream>
#include<algorithm>
using namespace std;
int main()
{
int n,m,i,j,a[1005],sum=0,temp;
cin>>n>>m;
for(i=0;i<n;i++)
cin>>a[i];
sort(a,a+n);
temp=n/m;
for(i=1;i<n/m;i++)
{
for(j=0;j<i*m;j++) //1 12
sum+=a[j];
}
for(i=n/m*m;i<n;i++)
{
for(j=i%m;j<n/m*m;j+=m)
sum+=a[j];
}
cout<<sum;
return 0;
}
//1 2 3
//4 5 6 1 2 3
//7 8 9 5 7 9
//10 11 12
//13 14 15