題目描述

　　給定一個正整數N，求最小的、比N大的正整數M，使得M與N的二進制表示中有相同數目的1。

　　舉個例子，假如給定的N為78，其二進制表示為1001110，包含4個1，那麽最小的比N大的並且二進制表示中只包含4個1的數是83，其二進制是1010011，因此83就是答案。

輸入格式

　　輸入若幹行，每行一個數n(1≤n≤1000000)，輸入"0"結束。

輸出格式

　　輸出若幹行對應的值。

輸入樣例

1

2

3

4

78

0

輸出樣例

2

4

5

8

83

題解

　　容易想到，當$n$加上$lowbit(n)$時，$1$的數量一定會減少$cnt$，我們只需要從末位開始找$cnt$個$0$位，將其改為$1$，得到的就是$m$了。

#include <iostream>
#include <cstdio>

#define lowbit(x) ((x) & -(x))

using namespace std;

int n, m;
int cnt;

int main()
{
    int tmp;
    while(scanf("%d", &n) && n)
    {
        m = n + lowbit(n);
        tmp = n;
        while(tmp) ++cnt, tmp -= lowbit(tmp);
        tmp 
 
= m;
        while(tmp) --cnt, tmp -= lowbit(tmp);
        tmp = 1;
        while(cnt)
        {
            if((tmp & lowbit(m)) ^ 1) --cnt, m += tmp;
            tmp <<= 1;
        }
        printf("%d\n", m);
    }
    return 0;
}

【題解】An Easy Problem