1. 程式人生 > >中等 更新二進位制位 (lintcode)

中等 更新二進位制位 (lintcode)

 更新二進位制位


給出兩個32位的整數N和M,以及兩個二進位制位的位置i和j。寫一個方法來使得N中的第i到j位等於M(M會是N中從第i為開始到第j位的子串)

樣例

給出N = (10000000000)2,M = (10101)2, i = 2, j = 6

返回 N = (10001010100)2

挑戰 

最少的操作次數是多少?


  1. 如果j<31,即j不在最高位上。可以把i到j位清為0,可以((1<<(j+1))-(1<<i))得到i到j之間全是1的數,再取反,得到i到j之間全是0的數。
  2. 如果j=32,(1<<(j+1))即(1<<32),相當於1<<1
     不可行。可以直接(1<<i)-1 得到i到j之間全是0,其他地方是1的數。
  3. 上面得到的數成為掩碼
  4. (m << i) + (n & mask)可以得到最終解。
class Solution {
public:
    /**
     *@param n, m: Two integer
     *@param i, j: Two bit positions
     *return: An integer
     */
    int updateBits(int n, int m, int i, int j) {
        // write your code here
        int mask;
        
        if(j < 31)
            mask = ~((1 << (j+1)) - (1 << i)  );
        else
            mask = (1 << i)-1;
        
        return (m << i)+(n & mask);
    }
};