05.01 插入

原题目

面试题 05.01. 插入

难度简单20收藏分享切换为英文接收动态反馈

插入。给定两个32位的整数N与M，以及表示比特位置的i与j。编写一种方法，将M插入N，使得M从N的第j位开始，到第i位结束。假定从j位到i位足以容纳M，也即若M = 10 011，那么j和i之间至少可容纳5个位。例如，不可能出现j = 3和i = 2的情况，因为第3位和第2位之间放不下M。

示例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中将[i, j+1]范围内清0，然后在[i, j+1]范围内插入M。

所以首先需要做一个mask，该mask在[i, j+1]范围为0，其余范围为1。

class Solution {
    public int insertBits(int N, int M, int i, int j) {
        int mask = (-1 << (j + 1)) | ~(-1 << i);
        return (N & mask) | (M << i);
    }
}

但是这儿有个隐藏的坑，java对于int左移实际应该是这样

int x = 1;
int i = 32;
x << i; // 1, 实际上是：x << (i % 32);

若 i 为32时，那么x << i为x本身。所以上述代码在j = 31时，可能会出现问题。更改代码如下：

class Solution {
    public int insertBits(int N, int M, int i, int j) {
        int mask = (-1 << j << 1) | ~(-1 << i);
        return (N & mask) | (M << i);
    }
}

小结

• 在Java中移动实际上为循环移动（左移和右移都是如此），要注意边界情况。