算法提高 打水问题
时间限制:1.0s 内存限制:512.0MB
问题描述
N个人要打水,有M个水龙头,第i个人打水所需时间为Ti,请安排一个合理的方案使得所有人的等待时间之和尽量小。
输入格式
第一行两个正整数N M 接下来一行N个正整数Ti。
N,M<=1000,Ti<=1000
输出格式
最小的等待时间之和。(不需要输出具体的安排方案)
样例输入
7 3
3 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
题意: 略
分析: 这是典型的贪心题,首先我们肯定把较大的放在后面,这样才能使得总的等待时间最小,我们可以利用vector,从小到大的依次放即可
参考代码
#include <bits/stdc++.h>
using namespace std;
int a[1234];
int main(){
int n,m;cin>>n>>m;
vector<vector<int> > b(m + 1,vector<int>());
for(int i = 0;i < n;i++) cin>>a[i];
sort(a,a+n);
for (int i = 0;i < n;i += m) {
for (int j = i;j < i + m && j < n;j++) {
b[j - i].push_back(a[j]);
}
}
int res = 0;
for (int i = 0;i < m;i++) {
int t = 0;
for (int j = 0;j < b[i].size() - 1;j++) {
t += b[i][j];
res += t;
}
}
cout<<res<<endl;
return 0;
}
- 如有错误或遗漏,请私聊下UP,thx