hihocoder 1269 - 优化延迟（二分 + heap）

原创于 2017-04-11 20:43:30 发布 · 292 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法

二分专栏收录该内容

27 篇文章

订阅专栏

本文介绍了一种使用二分查找结合优先队列解决特定问题的方法。通过定义合适的目标函数，利用二分查找确定最优缓冲区大小，并采用优先队列进行辅助验证，确保结果的有效性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目连接

https://hihocoder.com/problemset/problem/1269

思路

二分缓冲区的大小，在判断答案是否合法的时候，用堆去维护。

代码

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

#define LL long long
const int maxn = 100000 + 5;

LL N, Q;
LL p[maxn];

bool judge(int K) {
    LL res = 0, cnt = 1;
    priority_queue<int> pq;
    int i = 0;
    while (i < N && pq.size() < K) {
        pq.push(p[i]);
        i++;
    }
    while (pq.size()) {
        int x = pq.top(); pq.pop();
        res += cnt * x;
        cnt++;
        if (i < N) pq.push(p[i++]);
    }
    return res <= Q;
}

int main() {
    cin >> N >> Q;
    for (int i = 0; i < N; i++)
        scanf("%lld", &p[i]);
    LL l = 1, r = N, m, ans = -1;
    while (l <= r) {
        m = l + (r - l) / 2;
        if (judge(m)) r = m - 1, ans = m;
        else l = m + 1;
    }
    cout << ans << endl;
    return 0;
}