当前位置：首页 > news >正文

LeetCode 面试题 05.01. 插入

news 来源：原创 2025/4/22 16:00:59

文章目录

一、题目
二、Java 题解

一、题目

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