YBT 1223 An Easy Problem

输入若干个 n n n，对于每一个 n n n，输出最小的一个比它大 m m m，并且使它们在二进制下拥有相同的 1 1 1

我们可以这样处理每一个 n n n：先加上这个 n n n的 l o w b i t lowbit lowbit，然后再从最末位逐个添加 1 1 1直到 1 1 1的个数相等。以题目中的例子来说明， 78 78 78的二进制是 1001110 1001110 1001110，先加上它的 l o w b i t lowbit lowbit即 10 10 10得到 1010000 1010000 1010000，然后再逐个添加末位的 1 1 1使它 1 1 1的个数与原数一样得到 1010011 1010011 1010011即 83 83 83。

#include
#include
#define lb(x) ((x)&(-(x)))
#define ll long long
using namespace std;
ll n;
ll cot(ll x){
	ll sum=0;
	while(x){
		x-=lb(x);
		++sum;
	}
	return sum;
}
ll solve(ll x){
	ll res=lb(x),num=cot(x);
	x+=res;
	if(cot(x)==num)return x;
	ll i=1;
	while(cot(x)!=num){
		x+=i;
		i<<=1;
	}
	return x;
}
int main(){
	scanf("%lld",&n);
	while(n!=0){
		printf("%lldn",solve(n));
		scanf("%lld",&n);
	}
	return 0;
}