#yyds干货盘点# LeetCode程序员面试金典：下一个数

Posted 2022-12-30

题目：

下一个数。给定一个正整数，找出与其二进制表达式中1的个数相同且大小最接近的那两个数（一个略大，一个略小）。

示例1:

输入：num = 2（或者0b10）
 输出：[4, 1] 或者（[0b100, 0b1]）

示例2:

输入：num = 1
 输出：[2, -1]

代码实现：

class Solution 
    public int[] findClosedNumbers(int num) 
        int[] res =  new int[2];
        if (num <=0 || num>=Integer.MAX_VALUE) 
            res[0] = -1;
            res[1] = -1;
         else 
            res[0] = getNext(num);
            res[1] = getPrev(num);
        
        return res;
    
    
    // 取得后一个较大的数
    private int getNext(int n) 
        // 计算c0和c1，用于找到最右边非拖尾0的下标p
        int c = n;
        int c0 = 0;
        int c1 = 0;
        while (((c&1)==0)&&(c!=0)) 
            c0++;
            c >>= 1;
        
        while ((c&1)==1) 
            c1++;
            c >>= 1;
        
        
        // 错误：若n=111111...000, 那么就没有更大的数字
        // 如果是n的二进制不存在可翻转的0，或者n就是0
        if (c0 + c1 == 31 || c0 +c1 ==0) 
            return -1;
        
        
        int p = c0+c1; // 前提：最右边，非拖尾0的位置
        n |= (1<<p); // 步骤1：翻转最右边，非拖尾0
        n &= ~((1<<p)-1); // 步骤2：将p右方的所有位清零
        n |= (1<<(c1-1))-1; // 步骤3：在右方插入(c1-1)个1
        
        return n;