洛谷 P1582 倒水

题目描述
一天，CC买了N个容量可以认为是无限大的瓶子，开始时每个瓶子里有1升水。接着~~CC发现瓶子实在太多了，于是他决定保留不超过K个瓶子。每次他选择两个当前含水量相同的瓶子，把一个瓶子的水全部倒进另一个里，然后把空瓶丢弃。(不能丢弃有水的瓶子)
显然在某些情况下CC无法达到目标，比如N=3,K=1。此时CC会重新买一些新的瓶子(新瓶子容量无限，开始时有1升水)，以到达目标。
现在CC想知道，最少需要买多少新瓶子才能达到目标呢？
输入输出格式
输入格式：

一行两个正整数， N,K(1<=N<=10^9，K<=1000)。

输出格式：

一个非负整数，表示最少需要买多少新瓶子。

输入输出样例
输入样例#1：
　　　样例1：
3 1
　　　样例2：
13 2
　　　样例3：
1000000 5
输出样例#1：
　　　样例1：
1
　　　样例2：
3
　　　样例3：
15808

---

因为杯子里的水只可能是2^n，所以想到二进制。
题目为，把一个二进制数加一个尽可能小的数使他各个位置上1的个数不超过k个。

---

#include<iostream>
#include<cstdio>
using namespace std;
int cnt,s;
long long n,k,ans;
bool b[35];
long long pow(long long x)
{
    long long y=1;
    for(int i=1;i<=x;i++)
        y*=2;
    return y;
}
int main()
{
    scanf("%lld%lld",&n,&k);
    for(int i=31;i>=0;i--)
        if(n>=pow(i))
        {
            n-=pow(i);
            b[i]=1;
            cnt++;
        }
    while(cnt>k)
    {
        for(int i=s;;i++)
            if(b[i])
            {
                s=i;
                b[i]=0;
                break;
            }
        for(int i=s+1;;i++)
            if(b[i])
            {
                b[i]=0;
                --cnt;
                int x=i+1;
                while(b[x])
                {
                    b[x]=0;
                    x++;
                    --cnt;
                }
                b[x]=1;
                ans+=pow(i)-pow(s);
                s=x;
                break;
            }
    }
    printf("%lld\n",ans);
    return 0;
}