HDU5884（K叉哈夫曼树 + 二分）

题目链接

题意：给了你n个数，要把这n个数合成一个，每次可以拿k个，每次拿数的权值是所拿数之和，比如3 4 4三个数每次可以拿2个，那就是（3+4）+（3+4+4）。
求最后权值总和不超过m的最小k。

用2叉哈夫曼树疯狂tle，竟然还有k叉这种东西。
尽管在这个问题中k叉已经比2叉快了很多，但是还是需要二分枚举k的最小值，不然还是会超时。
k叉哈夫曼树大概就是用两个数组a，b来模拟哈夫曼树的过程，首先从a中取k个数求和存入b数组末尾中，然后从a和b数组中选前k小个数求和放入b数组，这样b数组一定是有序的，一直取到只剩下最后一个就可以了。
具体看这里：大佬传送门

#include<cstdio>
#include<cstring>
#include<queue>
#include<vector>
#include<algorithm>
using namespace std;
#define ll long long
int N;
ll a[100005],b[100005];
ll Hafuman(int k)  //返回总代价
{
    ll ai, bi, blen;
    blen = 0;
    ai = bi = 0;
    ll cost = 0;
    bool first = true;
    while (N - ai + blen - bi > 1)
    {
        ll num = 0;
        if (first)
        {
            if ((N - k) % (k - 1) == 0)
                num = k;
            else
                num = (N - k) % (k - 1) + 1;
            first = false;
        }
        else
            num = k;
        ll sum = 0;
        while (num--)
        {
            if (ai == N)
            {
                sum += b[bi];
                bi++;
            }
            else if (bi == blen)
            {
                sum += a[ai];
                ai++;
            }
            else if (a[ai] < b[bi])
            {
                sum += a[ai];
                ai++;
            }
            else
            {
                sum += b[bi];
                bi++;
            }
        }
        cost += sum;
        b[blen++] = sum;
    }
    return cost;
}
int main()
{
    int t,m,te = 1;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d %d",&N,&m);
        for (int i = 0; i < N; i++)
            scanf("%lld",&a[i]);
        sort(a,a + N);
        int l = 2,r = N;
        while(l < r)
        {
            int mid = (l + r)/2;
            if(Hafuman(mid) <= m)
            {
                r = mid;
            }
            else l = mid + 1;
        }
        printf("%d\n",l);
    }
}