HDU5884->贪心

5 篇文章 0 订阅

HDU5884->贪心(二分+单调队列)

题意:

n个有序序列的归并排序。每次可以选择不超过k个序列进行合并,合并代价为这些序列的权值和(类似果子合并).总的合并代价不能超过T, 问k最小是多少

题解:

贪心算法求解。
二分答案k,k可以从2取到n,由于每次将k个元素归并成了一个元素,也就是说归并一次序列长度减少了(k-1),最后要把n个序列归并成1个序列,相当于一共要减少的长度为(n-1).
计算mod=(n-1)%(k-1),如果mod==0,说明刚好k次能够归并完成,否则就利用贪心思想,先归并减少mod的序列长度,然后再每次归并k个。
在处理答案过程,可以开两个队列,一个存储原始的每个数的权值,一个存储合并后的权值,由于合并前先排过序,所以可以证明q2也一定单调。
//优先队列会TLE

代码:

#include <stdio.h>
#include <iostream>
#include <algorithm>
#include <queue>
using namespace std;
#define MAX 100005
int num[MAX];
queue<int>q1, q2, q3;
int T, ma, n;
int  temp, ans;
bool Judge(int mid)
{
    int mod;
    ans = 0;
    mod = (n - 1) % (mid - 1);
    if (!q1.empty()) q1.pop();
    if (!q2.empty()) q2.pop();
    q1 = q3;
    temp = 0;
    if (mod > 0)
    {
        for (int i = 0; i <= mod; i++)
        {
            temp += q1.front();
            q1.pop();
        }
        q2.push(temp);
        ans += temp;
        temp = 0;
    }
    while (q1.size() + q2.size() >= mid)
    {
        for (int i = 1; i <= mid; i++)
        {
            if (q1.empty())
            {
                temp += q2.front();
                q2.pop();
            }
            else if (q2.empty())
            {
                temp += q1.front();
                q1.pop();
            }
            else if (q1.front() > q2.front())
            {
                temp += q2.front();
                q2.pop();
            }
            else
            {
                temp += q1.front();
                q1.pop();
            }
        }
        ans += temp;
        q2.push(temp);
        temp = 0;
    }
    return ans <= ma;
}
int main()
{

    scanf("%d", &T);

    while (T--)
    {
        scanf("%d%d", &n, &ma);
        for (int i = 0; i < n; i++)
            scanf("%d", &num[i]);
        sort(num, num + n);
        while (!q3.empty())q3.pop();
        for (int i = 0; i < n; i++)
            q3.push(num[i]);
        int l = 2, r = n, mid;
        while (r - l>1)
        {
            mid = (l + r) / 2;
            if (Judge(mid)) r = mid;
            else l = mid;
        }
        printf("%d\n", (l == r || Judge(l)) ? l : r);
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值