HDU5884->贪心

来源:互联网 发布:知乎 mobi 编辑:程序博客网 时间:2024/06/05 17:08

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 100005int 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
原创粉丝点击