题目大意:
招收了n个拳击手,想让他们按照重量分成三类,每个都要符合2的幂次方。这n个人不能减少,但是可以在每组多增加招人,问最少需要招多少人。
解法:
我们简单思考可以直接二重枚举轻量级和中量级,直接枚举的时间复杂度大概是n(logn)(logn),对于2e5的n来说不会超时,很符合我们希望的时间。
于是代码如下
#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
int n;
int a[200005];
long long int solve(int w,int v){
int i=1;
long long res=0;
int sum=0;
while(sum+a[i]<=w){
sum+=a[i];
i++;
}
int u=n;
u-=sum;
res+=w-sum;
sum=0;
while(sum+a[i]<=v&&i<=n){
sum+=a[i];
i++;
}
u-=sum;
res+=v-sum;
for(i=0;(1<<i)<u;i++);
res+=(1<<i)-u;
return res;
}
int main(){
int t;
scanf("%d",&t);
while(t--){
scanf("%d",&n);
long long res=1e9+5;
for(int i=1;i<=n;i++)
a[i]=0;
for(int i=1;i<=n;i++){
int x;
scanf("%d",&x);
a[x]++;
}
for(int i=0;(1<<i)<=n;i++)
for(int j=0;(1<<j)<=n;j++)
res=min(res,solve((1<<i),(1<<j)));
printf("%lld\n",res);
}
}