试题 算法提高 排队打水问题
资源限制
内存限制:256.0MB C/C++时间限制:1.0s Java时间限制:3.0s Python时间限制:5.0s
问题描述
有n个人排队到r个水龙头去打水,他们装满水桶的时间t1、t2…………tn为整数且各不相等,应如何安排他们的打水顺序才能使他们总共花费的时间最少?
输入格式
第一行n,r (n<=500,r<=75)
第二行为n个人打水所用的时间Ti (Ti<=100);
输出格式
最少的花费时间
样例输入
3 2
1 2 3
样例输出
7
数据规模和约定
其中80%的数据保证n<=10
分析
总共花费的时间 = 每个人的打水时间 + 等待时间 的总和
切入点
打水时间已经规定,要想总花费时间最短,只能尽力减少等待时间
即,让打水快的先打,开敲
思路
将打水的人进行时间升序,保存在一个数组里
将水龙头保存在数组里,记录当前队列打水需要多长时间,即下一个人打水需要等待的时间
当一个人进入等待队列后,累计总花费时间
,同时队列会重新排序,确保下一个人进入到最短的等待队列,其实就是搜索数组中的最小值。为了代码方便用排序代替,确保第1个队列永远最短。
代码
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
// 封装一个水龙头结构体
struct {
int timeLong;
}typedef WaterHead;
bool mysort(WaterHead a, WaterHead b) {
return a.timeLong < b.timeLong;
}
int main() {
int n, r; // 人数,水龙头数目
cin >> n >> r;
// 准备水龙头
vector<WaterHead> wh(r);
vector<int> ts(n);
for (int i = 0; i < n; i++) {
cin >> ts[i];
}
sort(ts.begin(), ts.end());
int res = 0;
for (int i = 0; i < n; i ++) {
// 进入队列,添加队列总时间
wh[0].timeLong += ts[i];
res += wh[0].timeLong;
// 排序水龙头(升序)
sort(wh.begin(), wh.end(), mysort);
}
cout << res << endl;
return 0;
}