Problem E: 倒水(Water)
Description
一天,CC买了N个容量可以认为是无限大的瓶子,开始时每个瓶子里有1升水。接着~~CC发现瓶子实在太多了,于是他决定保留不超过K个瓶子。每次他选择两个当前含水量相同的瓶子,把一个瓶子的水全部倒进另一个里,然后把空瓶丢弃。(不能丢弃有水的瓶子)
显然在某些情况下CC无法达到目标,比如N=3,K=1。此时CC会重新买一些新的瓶子(新瓶子容量无限,开始时有1升水),以到达目标。
现在CC想知道,最少需要买多少新瓶子才能达到目标呢?
Input
第一行一个整数T,表示有T组数据。
接着T行,每行两个正整数, N,K(1<=N<=10^9,K<=1000)。
Output
一个非负整数,表示最少需要买多少新瓶子。
Sample Input
33 113 21000000 5
Sample Output
1315808
把N写成二进制的形式就什么都明了了
// by BNU_LZM
#include<cstdio>
#include<cstring>
int a[32];
int solve(int n, int k)
{
memset(a, 0, sizeof(a));
int cnt = 0;
while(n)
{
a[cnt++] = n%2;
n /= 2;
}
int cnt1 = 0;
for(int i = 0; i < cnt; i++) if(a[i]) cnt1++;
if(k >= cnt1) return 0;
int ans = 0;
cnt1 = 0;
for(int i = cnt-1; i >= 0; i--)
{
if(a[i]) cnt1++;
if(cnt1 == k)
{
cnt1 = i;
break;
}
}
int ok = 0, x;
for(int i = 0; i <= cnt1; i++)
{
if(!ok && a[i])
{
ok = 1;
x = i;
}
else if(ok && a[i])
{
ans += ((1<<i) - (1<<x));
x = i+1;
}
}
return ans;
}
int main()
{
freopen("in.txt", "r", stdin);
int T;
scanf("%d", &T);
while(T--)
{
int n, k;
scanf("%d%d", &n, &k);
printf("%d\n", solve(n, k));
}
return 0;
}