题意:
给定数组,可选 \(x<y\),把小于 \(x\) 的数划为div1,除此之外小于 \(y\) 的数划为div2,剩下的数为div3。现往每组中放入额外的数,使得每组的大小都是2的幂。输出至少要加几个数。
n <= 2e5,1 <= ai <= n
思路:
开桶记录出现次数。枚举加数后div1的大小 \(2^i\) 以及加数后div2的大小 \(2^j\),从左到右尽量填满 \(2^i\),再从左到右尽量填满 \(2^j\) (二分找区间端点),剩下的属于div3。
const int N = 2e5 + 5, M = 18, INF = 0x3f3f3f3f;
int n, s[N];
int cal(int x) { //计算x至少加多少才是2的幂
for(int i = 0; ; i++) if((1<<i) >= x)
return (1<<i) - x;
}
int work(int p1, int p2) { //区间划分为[1,p1],[p1+1,p2],[p2+1,n]
return cal(s[p1]) + cal(s[p2]-s[p1]) + cal(s[n]-s[p2]);
}
void sol()
{
scanf("%d", &n); fill(s, s + n + 1, 0);
for(int i = 1, x; i <= n; i++) scanf("%d", &x), s[x]++;
for(int i = 1; i <= n; i++) s[i] += s[i-1];
int ans = INF;
for(int i = 0; i < M; i++)
for(int j = 0; j < M; j++)
{
int r1 = (1<<i);
int p1 = upper_bound(s+1,s+1+n,r1) - s - 1;
int r2 = s[p1] + (1<<j);
int p2 = upper_bound(s+1,s+1+n,r2) - s - 1;
ans = min(ans, work(p1,p2));
}
printf("%d\n", ans);
}