hdu 5884 二分 + 前k优化

最新推荐文章于 2021-04-02 17:27:31 发布

羁绊残阳

最新推荐文章于 2021-04-02 17:27:31 发布

阅读量400

点赞数

分类专栏： ACM_技巧

本文链接：https://blog.csdn.net/jibancanyang/article/details/52572933

版权

ACM_技巧专栏收录该内容

31 篇文章 0 订阅

订阅专栏

分析：

大多数人都可以直接想到二分，然后用优先队列维护每次取前 $k$ 大的合并。
但是这样复杂度多了一个 $logn$ 会TLE。这里想到一个很好的性质，就是合并之后的数，一定比之前合并的数大，所有就把合并之后的数加入一个队列。对原数组也排序处理，这样就得到了两个有序数组，每次在头尾取前 k <script type="math/tex" id="MathJax-Element-66">k</script>大即可。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define pr(x) cout << #x << ": " << x << "  " 
#define pl(x) cout << #x << ": " << x << endl;
const int maxn = int(1e5) + 12;


struct jibancanyang
{
    int A[maxn], n;
    int B[maxn];
    long long sum;


    bool judge(int k) {
        long long cost = 0;
        int p = 1;
        int head = 1, tail = 1;
        int mod = (n - 1) % (k - 1) + 1;
        for (int i = 0; i < mod; ++i) {
            cost += A[p++];
        }
        if (cost) B[tail++] = cost;
        while (true) {

            long long  temp = 0;
            for (int i = 0; i < k; ++i) {
                if (p <= n && ( head >= tail || A[p] <= B[head]) ) temp += A[p++];
                else if (head < tail && (p > n || B[head] <= A[p])) temp += B[head++];
            }
            cost += temp;
            if (p > n && head >= tail) break;
            B[tail++] = temp;
        }
        return cost <= sum; 
    }


    void fun()  {
        int T;
        scanf("%d", &T);
        while (T--) {
            scanf("%d%lld", &n, &sum);
            for (int i = 1; i <= n; ++i) {
                scanf("%d", &A[i]);
            }
            int l = 2, r = n;
            sort(A + 1, A + 1 + n);
            while (l < r) {
                int mid = (l + r) / 2;
                if (judge(mid)) r = mid;
                else l = mid + 1;
            }
            printf("%d\n", l);
        }
    }

}ac;

int main()
{
#ifdef LOCAL
    freopen("in.txt", "r", stdin);
    //freopen("out.txt", "w", stdout);
#endif
    ac.fun();
    return 0;
}