[蓝桥杯][算法提高][vip] 打水问题

版权声明：原创博客，转载请标明出处！ https://blog.csdn.net/caipengbenren/article/details/88596334

蓝桥杯 ADV-104 算法提高 打水问题

问题描述
　　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

---

这是一道贪心排序的题目，和之前那个题目很类似。读者可自行比较区别，方法基本一样，细节处理上有所不同。
排队打水问题

---

#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;
int tim[1001];
int main() {
    int n, m;
    cin >> n >> m;
    for (int i = 1; i <= n; i++) {
        cin >> tim[i];
    }
    sort(tim + 1, tim + n + 1);
    queue<int > que;
    for (int i = 1; i <= m; i++) {
        que.push(tim[i]);
    }
    int index = m + 1, ans = 0;
    while(index <= n) {
        ans +=que.front();
        que.push(tim[index++] + que.front());
        que.pop();
    }
    printf("%d", ans);
    return 0;
}