算法提高 排队打水问题

试题 算法提高 排队打水问题

资源限制

内存限制：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;
}