问题描述
有n个人排队到r个水龙头去打水,他们装满水桶的时间t1、t2…………tn为整数且各不相等,应如何安排他们的打水顺序才能使他们总共花费的时间最少?
输入格式
第一行n,r (n<=500,r<=75)
第二行为n个人打水所用的时间Ti (Ti<=100);
输出格式
最少的花费时间
样例输入
3 2
1 2 3
样例输出
7
思路解析
- 按照时间从小到大的顺序排序后即为打水的顺序
- 讲打水时间储存在数组中,并且用 vector 循环储存 r 次
- 算出每个 vector 中的时间和
一种最佳打水方案是,将N个人按照Ti从小到大的顺序依次分配到M个龙头打水。
例如样例中,Ti从小到大排序为1,2,3,4,5,6,7,将他们依次分配到3个龙,则
去龙头一打水的为 1,4,7;
去龙头二打水的为 2, 5;
去龙头三打水的为 3,6。
第一个龙头打水的人总花费时间 = 13+42+71=18
第二个龙头打水的人总花费时间 = 22+51=9
第三个龙头打水的人总花费时间 = 32+6*1=12
所以总的等待时间 = 18+9+12=39
代码
#include <iostream>
#include <math.h>
#include <string>
#include <vector>
#include <algorithm>
using namespace std;
int main()
{
int n,r,sum=0;
cin>>n>>r;
int a[1000];
for(int i=0;i<n;i++)
cin>>a[i];
for(int j=0;j<r;j++)
{
int i=0;
vector<int> b;
while(j+i<n)
{
b.push_back(a[j+i]);
i+=r;
}
int x=b.size();
for(int i=0;i<b.size();i++)
{
sum+=b[i]*x;
x-=1;
}
}
cout<<sum<<endl;
}
输出结果