LeetCode 面试题 05.01. 插入

一、题目

  给定两个整型数字 NM,以及表示比特位置的 ij(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;
    }
}
  • 时间:0 ms,击败 100.00% 使用 Java 的用户
  • 内存:36.93 MB,击败 91.67% 使用 Java 的用户

猜你喜欢

转载自blog.csdn.net/zheliku/article/details/133171976