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