hdu 5884- Sort 哈夫曼树构造 + 模拟优先队列 + 二分

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