下一个数。给定一个正整数,找出与其二进制表达式中1的个数相同且大小最接近的那两个数(一个略大,一个略小)。
示例1:
输入:num = 2(或者0b10)
输出:[4, 1] 或者([0b100, 0b1])
示例2:
输入:num = 1
输出:[2, -1]
提示:
num
的范围在[1, 2147483647]之间;- 如果找不到前一个或者后一个满足条件的正数,那么输出 -1。
本题知识点在于位运算,熟悉位运算能较快解题。
- 如何取大一点的数:从右向左遍历num二进制表达式,将第一次出现 “01” 的位置反转成 “10” ,“***01###”转换成“***10###”,并且“###”字段的“1”全部聚集在尾部
- 如何取小一点的数:从右向左遍历num二进制表达式,将第一次出现**“10”的位置反转成“01”**,“***10###”转换成“***01###”,并且“###”字段的“1”全部聚集在头部
#include <iostream>
#include<vector>
using namespace std;
class Solution {
public:
vector<int> findClosedNumbers(int num) {
vector<int> ans(2,-1);
//大一点的数
int cnt=0;
for(int i=0;i<30;i++){
if(((num>>i)&1)==1 && ((num>>(i+1))&1)==0){
ans[0] = ((num|(1<<(i+1)))^(1<<i))-(num&((1<<i)-1))+((num&((1<<i)-1))>>cnt);
break;
}
if(((num>>i)&1)==0) cnt++;
}
//小一点的数
cnt=0;
for(int i=0;i<31;i++){
if(((num>>i)&1)==0 && ((num>>(i+1))&1)==1){
ans[1] = ((num|(1<<i))^(1<<(i+1)))-(num&((1<<i)-1))+((num&((1<<i)-1))<<cnt);
//printf("%d %d %d %d\n",((num|(1<<i))^(1<<(i+1))),(num&((1<<i)-1)),((num&((1<<i)-1))<<cnt),cnt);
break;
}
if(((num>>i)&1)==0) cnt++;
}
return ans;
}
};
位运算操作tip:
-
确定num第i位是否为1
(num>>i)&1
-
截取num的后面i位
num&((1<<i)-1)
-
注意位运算与其他运算的优先级
加减乘除(+,-,*,/) > 位移(<<,>>) > 大于小于(=,>,<) > 按位与(&,|,^) > 逻辑与(&&,||)