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
挑战
最少的操作次数是多少？

solution: 题目的意思就是讲N的第i到j位换成M，因此，我们可以先将N的i到j位清零和把M向右移动i位，然后将N与上M就是所求的答案。

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
        for (int pos = i; pos <= j; ++pos)
            n &= ~(1<<pos);
        m <<= i;
        return n | m;
    }
};

其时间复杂度为$O(j-i)$, 当然我们也可以在$O(1)$的时间内完成。、

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 ones = ~0;  //将所有的位置为1
        int mask;
        if (j < 31) {
            mask = ones << (j+1);   //将0到j位都置为0
            mask |= ((1<<i) - 1);    //将i到j位置为0，其余位为1;
        } else {
            mask = ((1<<i) - 1);    //此时j = 32;
        } 

        return (n & mask) | (m<<i) ;
    }
};