LeetCode 面试题 05.01. 插入

一、题目

  给定两个整型数字 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)

二、Java 题解

  将 N 分为 3 个区间：[31, j + 1] 为前缀 pre；[j, i] 为中间 mid；[i, 0] 为后缀 suf。前缀 pre 使用除法求解，后缀 suf 使用除余求解。需要注意的是，求解 pre 时，j = 31 会产生 int 范围越界，因此直接赋值为 0 即可。

class Solution {
    public int insertBits(int N, int M, int i, int j) {
        int mid = M << i; // 中间
        int pre = j == 31 ? 0 : (N / (1 << (j + 1))) << (j + 1); // 前缀
        int suf = N % (1 << i); // 后缀

        return pre | mid | suf;
    }
}
1
2
3
4
5
6
7
8
9

• 时间：0 ms，击败 100.00% 使用 Java 的用户
• 内存：36.93 MB，击败 91.67% 使用 Java 的用户