武汉ACM集训——贪心-4

武汉ACM集训——贪心-4

洛谷 P1190 [NOIP2010 普及组] 接水问题

题解

按照顺序排队接水，一个接完水，另一个瞬间接上。

将前 m 个数据作为 m 组，选取时间和最小的组，再该组数据中添加新数据，重复添加新数据。输出最大的时间和。

AC代码

#include <bits/stdc++.h>
using namespace std;

const int maxn = 1e4 + 5;
int a[maxn];
// 优先队列，数值越小优先级越高
priority_queue<int, vector<int>, greater<int>> b;

int main()
{
    int n, m;
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; ++i)
        scanf("%d", &a[i]);
    // 前 m 个数据成组
    for (int i = 1; i <= m; ++i)
    {
        b.push(a[i]);
    }
    // 每次选取最小的组添加新数据
    int tmp = 0;
    for (int i = m + 1; i <= n; ++i)
    {
        tmp = b.top();
        tmp += a[i];
        b.pop();
        b.push(tmp);
    }
    // 输出最大的组
    for (int i = 1; i < m; ++i)
        b.pop();
    tmp = b.top();
    printf("%lld", tmp);
    return 0;
}