hdu 5884

首先将一下我的思路历程:
读题之后发现,有n-1个数,每次扩展k-1个数.
刚开始想的是直接用队列,但是发现合并之后的数字不一定是新序列里面最小的那一个,所以就考虑到了优先队列,但是这里有一个坑点,就是如果我们不能刚好把序列里面的这些数扩展完(即最后一次扩展并不是k-1个数),那么我们需要先进行一次扩展,使剩下的数在扩展时满足每一次都被扩展k-1个数的条件,这里很多博客并没有见讲很清楚,我看了下哈夫曼树后发现如果把这些小的数先进行一次合并的话,那么比最后进行小于k-1个数的合并要优化.
还有个就是卡时间的问题,如果我们加一个前缀和就不会超时,就是判断如果最后一次扩展不是k-1个数,那么让这个res直接加上前面数字的前缀和,感觉有点运气成分,但确实过题了0.0
代码如下:

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=1e5+100;
int arr[N],t,n,T;
int sum[N];
priority_queue<int,vector<int>,greater<int> >q;
bool ok(int k)
{
    int cc=(n-1)%(k-1);
    ll res=0,ans=0;
    if(cc)
    {
        cc++;
        res+=sum[cc];
        q.push(sum[cc]);
    }
    for(int i=cc+1;i<=n;i++)
        q.push(arr[i]);
    int cnt=(n-1)/(k-1);
    while(cnt--)
    {
        ans=0;
        int m=k;
        while(m--)
        {
            ans+=q.top();
            q.pop();
        }
        q.push(ans);
        res+=ans;
    }
    if(res<=T)
        return 1;
    else
        return 0;
}
int main()
{
    ios::sync_with_stdio(false);
    cin>>t;
    while(t--)
    {
        cin>>n>>T;
        for(int i=1;i<=n;i++)
            cin>>arr[i];
        sort(arr+1,arr+1+n);
        sum[0]=0;
        for(int i=1;i<=n;i++)
            sum[i]=sum[i-1]+arr[i];
        int l=2,r=n;
        while(l<=r)
        {
            while(!q.empty())
                q.pop();
            int mid = (l+r)/2;
            if(ok(mid))
                r=mid-1;
            else
                l=mid+1;
        }
        cout<<l<<endl;
    }
    return 0;
}