Problem E: 倒水(Water)

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

这个题模拟就可以，一开始不断的消掉瓶子，如果期间有出现奇数的情况，就把瓶子留下，最后一直消直到剩下的瓶子数>=k，结束。

然后把剩下的瓶子按照水的容量排序，从小到大不断消，直到最后剩下k个就OK了。

#include <bits/stdc++.h>
 
using namespace std;
const int MAXN=1e5+7;
const int mod=10007;
long long ans[MAXN];
int n,m,k;
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int cnt=0;
        scanf("%d%d",&n,&k);
        if(n<k)printf("%d\n",n-k);
        else
        {
            int flag=0;
            long long p=1;
            while(n)
            {
                if(cnt+n==k)
                {
                    flag=1;
                    break;
                }
                if(cnt+n%2+(n>>1)<k)break;
                if(n%2)ans[cnt++]=p;
                p<<=1;
                n>>=1;
            }
            if(flag)puts("0");
            else
            {
                for(int i=0;i<n;++i)
                {
                    ans[cnt++]=p;
                }
                int l=cnt-k;
                long long sum=0;
                for(int i=0;i<l;++i)
                {
                    sum+=ans[i+1]-ans[i];
                    ans[i+1]<<=1;
                }
                printf("%lld\n",sum);
            }
        }
    }
    return 0;
}