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;
}