给定两个整型数字 N 与 M,以及表示比特位置的 i 与 j(i <= j,且从 0 位开始计算)。
编写一种方法,使 M 对应的二进制数字插入 N 对应的二进制数字的第 i ~ j 位区域,不足之处用 0 补齐。具体插入过程如图所示。
题目保证从 i 位到 j 位足以容纳 M, 例如: M = 10011,则 i~j 区域至少可容纳 5 位。
示例1:
输入:N = 1024(10000000000), M = 19(10011), i = 2, j = 6
输出:N = 1100(10001001100)
示例2:
输入: N = 0, M = 31(11111), i = 0, j = 4
输出:N = 31(11111)
分析:
方法:位运算
题目要求替换掉 N 的中间部位,那么我们可以把 N 拆分为左边、中间、右边部分,空位补 0,将 M 左移 i 位就是新的中间部分,和 N 的左边右边做或运算就可以实现替换操作。
左边部分可以通过先将 N 右移 j + 1 为再左移 j + 1 位实现 j 以及 j 以前清零。
右边部分可以通过先对 1 左移 i 位减 1,这样在 i 位之前都为 1,用该数和 N 做与运算就可以实现 i 以及 i 以后清零。
时间复杂度:O(1)
空间复杂度:O(1)
class Solution {
public int insertBits(int N, int M, int i, int j) {
//左边
int left = N >> j >> 1 << 1 << j;
//右边
int right = N & (1 << i) - 1;
return left | right | (M << i);
}
}
题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/insert-into-bits-lcci