SSLOJ·杯子【模拟】

Description–

小明买了N个容积可以是无穷大的杯子，刚开始的时候每个杯子里有1升水，接着小明发现杯子实在太多了，于是他决定保留不超过K个杯子。每次他选择两个当前含水量相等的杯子，把一个杯子的水全部倒进另一个里，然后把空瓶丢弃。（不能丢弃有水的杯子）
显然在有些情况下小明无法达到他的目标，比如N=3，K=1。此时小明会重新买一些新的杯子（新杯子容积无限，开始时有1升水），以达到目标。
现在小明想知道，最少需要买多少个新杯子才能达到目标呢？

---

Input–

一行两个正整数，N，K（1≤N≤1000000000，K≤1000）。

Output–

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

---

Sample Input–

【样例输入1】

3 1

【样例输入2】

13 2

【样例输入3】

1000000 5 

Sample Output–

【样例输出1】

1

【样例输出2】

3

【样例输出3】

15808

---

说明–

【数据范围】
对于50%的数据，N≤10000000；
对于100%的数据如题目。

---

解题思路–

a[ i ]的 i 表示 2^i
a[ i ]表示装有 2^i 升水的杯子数量
因为新加的杯子里只有1升水，所以我们需要几升水就是要几个杯子

---

代码–

#include<iostream>
#include<cstdio>
using namespace std;
long long s,t,k,ans,a[35],b[35];
void hhh()
{
	b[0]=1;
	for (int i=1;i<35;++i)
	  b[i]=b[i-1]*2;
}
int main()
{
	scanf("%lld%lld",&a[0],&k);
	while (a[t]>1)
	{
		a[++t]=a[t-1]/2;
		a[t-1]%=2;
		s+=a[t-1]; 
	}
	s+=a[t];
	if (s<=k)
	{
		printf("0");
		return 0;
	}
	hhh();
	for (int i=0;i<35;++i)
	{
		if (a[i]==2) a[i]=0,a[i+1]++,s--;
		if (a[i]==1) a[i]=0,ans+=b[i],a[i+1]++;
		if (s<=k) break;
	}
	printf("%lld",ans);
	
	return 0;
}