http://acm.hdu.edu.cn/showproblem.php?pid=5884
题意:模拟归并的过程,每次可以归并k个.要求最后总花费不超过T
这个花费就是哈夫曼树的花费...
我们二分k,问题变成了给定n,然后跑k叉哈夫曼树。
如果n不是整k倍,那么
“于是对给定的n个权值构造k叉哈夫曼树时,可以先考虑增加一些权值为0的叶子节点,使得叶子节点总数为(n-1)%(k-1)+1这种形式”,表示有这么多个是不能取完最后一次剩下的,(要消掉n-1个,每次消掉k-1个)
然后就可以了
由于队列B的单调性,可以用队列模拟优先队列,省去一个log
#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <iostream>
using namespace std;
const double pi=acos(-1.0);
double eps=0.000001;
typedef long long ll;
const int N=100000+50;
ll aa[N];
queue<ll> q;
int n;
ll T;
ll bin(int k)
{
long long cost=0;
while(!q.empty())q.pop();
int p=1;
for (int i=1; i<=(n-1)%(k-1)+1; i++)
cost+=aa[p++];
q.push(cost);
while(1)
{
if (q.empty()&&p>n) break;
ll tmp=0;
for(int i=0; i<k; i++)
{
if(q.empty())
{
tmp+=aa[p++];
}
else if(p>n)
{
tmp+=q.front();
q.pop();
}
else
{
ll val=q.front();
if (val<aa[p])
{
tmp+=val;
q.pop();
}
else tmp+=aa[p++];
}
if (q.empty()&&p>n) break;
}
cost+=tmp;
if (q.empty()&&p>n) break;
q.push(tmp);
}
return cost;
}
int main()
{
int t;
cin>>t;
while(t--)
{
scanf("%d%lld",&n,&T);
for (int i=1; i<=n; i++)
scanf("%lld",&aa[i]);
sort(aa+1,aa+1+n);
int l=2,r=n,ans=-1;
while(l<=r)
{
int mid=(l+r)>>1;
if (bin(mid)<=T)
r=mid-1,ans=mid;
else
l=mid+1;
}
printf("%d\n",ans);
}
return 0;
}