An Easy Problem

GitHub跳转原题关系图返回列表

求最小的比 n 大、且二进制中 1 的个数与 n 相同的数:对最低位 1 做一次 Gosper 进位,再把被清空的 1 补回最低位,单次 O(1)。

OJ: roj

题目 ID: 1223

难度:普及-

标签:位运算贪心python

创建: 2026-09-30 00:29

更新: 2026-10-05 05:57

题目描述

给定一个正整数 NN,求最小的、比 NN 大的正整数 MM,使得 MM 与 NN 的二进制表示中有相同数目的 1。

例如 N=78=(1001110)2N = 78 = (1001110)_2 有 4 个 1,满足条件的最小数是 83=(1010011)283 = (1010011)_2。

输入若干行,每行一个数 nn(1≤n≤1061 \le n \le 10^6),输入 0 结束。对每个 nn 输出一行对应的值。

样例:输入 1 / 2 / 3 / 4 / 78 / 0,对应输出 2 / 4 / 5 / 8 / 83。

思路

把 nn 最低位那段连续的 1 整体进位一格(low = n & -n,carried = n + low),就完成了"变大"且 popcount 少了若干个 1;再把被清空的 1 补回最低位即可最小。被改动位为 n ^ carried,右移两位丢掉新产生的 1,再除以 low 折算到最低位,得到要补回的 1 串,按位或回去即为答案,单次询问 O(1)O(1)。

参考代码

Python 版:

python
#!/usr/bin/env python3
# Author by Rainboy blog: https://rainboylv.com github: https://github.com/rainboylvx
# rbook: -> https://rbook.roj.ac.cn  https://rbook2.roj.ac.cn
# rainboy的学习导航网站: https://idx.roj.ac.cn
# create_at: 2026-09-30 00:29
# update_at: 2026-09-30 00:29

import sys
from itertools import takewhile


def next_same_ones(n: int) -> int:
    """返回大于 n 的最小整数,其二进制表示中 1 的个数与 n 相同(Gosper 进位法)。"""
    low = n & -n                                  # n 最低位置 1 的值 2^l,即进位的入口
    carried = n + low                             # 从第 l 位起的连续 1 全部进位清空,其上方第一个 0 变 1
    return carried | ((n ^ carried) >> 2) // low   # 把被清空的 k 个 1 中的 k-1 个铺回最低位


def solve() -> None:
    data = map(int, sys.stdin.buffer.read().split())
    out = [str(next_same_ones(n)) for n in takewhile(bool, data)]  # 读到 0(假值)即结束
    sys.stdout.write('\n'.join(out))


if __name__ == "__main__":
    solve()

C++ 版(同一算法):

cpp
/**
 * Author by Rainboy blog: https://rainboylv.com github: https://github.com/rainboylvx
 * rbook: -> https://rbook.roj.ac.cn  https://rbook2.roj.ac.cn
 * rainboy的学习导航网站: https://idx.roj.ac.cn
 * create_at: 2026-10-05 05:57
 * update_at: 2026-10-05 05:57
 */
#include <bits/stdc++.h>

typedef long long ll;

// 求比 n 大、且二进制中 1 的个数与 n 相同的最小数(Gosper 进位)
ll next_same_ones(ll n) {
    ll low = n & -n;      // 最低位的 1,即 2^l
    ll carried = n + low; // 最低位加 1 引发连锁进位:连续 1 段清空,上方第一个 0 位变 1
    // n ^ carried 的置 1 位是被改动的位;>> 2 丢掉新产生的那位,// low 把剩下的 1 折算到最低位
    return carried | (((n ^ carried) >> 2) / low);
}

int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(NULL);

    ll n;
    while (std::cin >> n && n != 0) { // 输入以 0 结束
        std::cout << next_same_ones(n) << "\n";
    }
    return 0;
}