2018TYUT暑期ACM模拟赛(7)
​​​Sort HDU - 5884​​​
题意:让你合并n个有序序列,合并k个有序序列的花费为k个序列的长度之和,问合并n个序列在不大于T的花费情况下。每次选择k个序列进行合并的k最小为多少。
思路:非常像哈夫曼树,然后就想着k叉哈夫曼树。找了一个k叉哈夫曼树的模板返回的是构造这个树的代价。然后二分找k的值。如果代价ans大于T,说明这个二分的k值小了,反之大了。

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=100000+10;
ll a[maxn];
ll b[maxn];
ll N,M;
ll Hafuman(int k) //返回总代价
{
int ai,bi,blen;
blen=0;//b数组的长度
ai=bi=0;
ll cost=0;
bool first=true;
while (N-ai+blen-bi>1)
{
int num=0;
if (first) //先取出 取不完k的余值x
{
if((N-k)%(k-1)==0) num=k;
else num=(N-k)%(k-1)+1;
first=false;
}
else num=k;
int sum=0;//本次合并
while(num--)//从b a中取出num个数相加 放入b数组
{
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;
if(cost>M) return M+1;
}
return cost;
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
scanf("%lld%lld",&N,&M);
for(int i=0;i<N;i++)
{
scanf("%d",&a[i]);
}
sort(a,a+N);
int l=1,r=N;
while(l<r)
{
int mid=(l+r)/2;
ll tmp=Hafuman(mid);
if(tmp<=M)
{
r=mid;
}
else
l=mid+1;
}
printf("%d\n",r);
}
return 0;
}