华为OD机试C卷-找数字(Python)

题目描述
小扇和小船今天又玩起来了数字游戏，小船给小扇一个正整数n(1≤n≤1e9)，小扇需要找到一个比n大的数字m，使得m和n对应的二进制中1的个数要相同，如：4对应二进制100，8对应二进制1000，其中1的个数都为1个，现在求m的最小值。
输入描述
输入一个正整数n(1≤n≤1e9)
输出描述
输出一个正整数m

Python实现：

# 计算一个数的二进制表示中1的个数
def count_ones(num):
    count = 0
    while num:
        count += num & 1  # 每次判断最低位是否为1
        num >>= 1  # 将数字右移一位，相当于除以2
    return count

# 找到比给定数字n大的下一个数字m，使得m和n对应的二进制中1的个数相同
def find_next_number(n):
    ones_count = count_ones(n)  # 统计n中1的个数
    m = n + 1  # 初始m为n加1
    while count_ones(m) != ones_count:  # 当m的二进制表示中1的个数不等于n的时候
        m += 1  # 继续增加m
    return m

n = int(input())  # 从标准输入中读取整数n
result = find_next_number(n)  # 调用函数找到满足条件的m
print(result)  # 打印结果m