中等 更新二进制位 (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

挑战 

最少的操作次数是多少？

标签 

Cracking The Coding Interview  比特位操作

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);
    }
};