YBT 1223 An Easy Problem

题目描述

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

题目分析

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

Code

#include<iostream>
#include<cstdio>
#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("%lld\n",solve(n));
		scanf("%lld",&n);
	}
	return 0;
}