蓝桥杯 ADV-148 排队打水问题

算法提高 排队打水问题  

时间限制：1.0s   内存限制：256.0MB

    

问题描述

　　有n个人排队到r个水龙头去打水，他们装满水桶的时间t1、t2………..tn为整数且各不相等，应如何安排他们的打水顺序才能使他们总共花费的时间最少？

输入格式

　　第一行n，r (n<=500,r<=75)
　　第二行为n个人打水所用的时间Ti (Ti<=100)；

输出格式

　　最少的花费时间

样例输入

3 2
1 2 3

样例输出

7 

数据规模和约定

　　其中80%的数据保证n<=10

分析：每个人在水龙头前排队的时间等于自己打水的时间和自己之前所有人打水的时间和。为了取得最少的时间，将打水时间少的人排在队伍前。最终求出每一条队伍的总时间和。

代码：

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
int main() {
	int n, r;
	cin >> n >> r;
	vector<int> v(n);
	for (int i = 0; i < n; i++) {
		cin >> v[i];
	}
	sort(v.begin(), v.end());
	vector<vector<int> > R(r);
	int len = r < n ? r : n;
	for (int i = 0; i < len; i++) {
		R[i].push_back(v[i]);
	}
	for (int i = len; i < n; i++) {
		R[(i - len) % r].push_back(v[i] + R[(i - len) % r][R[(i - len) % r].size() - 1]);
	}
	int ans = 0;
	for (int i = 0; i < R.size(); i++) {
		for (int j = 0; j < R[i].size(); j++) {
			ans += R[i][j];
		}
	}
	cout << ans << endl;
	return 0;
}