#179 Update Bits

题目描述：

Given two 32-bit numbers, N and M, and two bit positions, iand j. Write a method to set all bits between i and j in N equal to M (e g , M becomes a substring of N located at i and starting at j)

 Notice

In the function, the numbers N and M will given in decimal, you should also return a decimal number.

Have you met this question in a real interview? 

Yes

Clarification

You can assume that the bits j through i have enough space to fit all of M. That is, if M=10011， you can assume that there are at least 5 bits between j and i. You would not, for example, have j=3 and i=2, because M could not fully fit between bit 3 and bit 2.

Example

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

return N=(10001010100)2

Challenge 

Minimum number of operations?

解题思路：

这题的思路应该很多。我想到的一种比较容易想清楚的解法是：先把N中i到j位的bit清零（这个操作可以用一个mask做到），再加上M*2^i即可。这题比较容易出错的地方是要考虑到m或n < 0的情况，这种情况下不能直接做bit operation，应该先用long long的type将m或n存起来，如果<0则加上2^32，改成正数再做上述操作。

Mycode（AC = 22ms）：

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
        long long long_n = n, long_m = m;
        if (long_n < 0) {
            long_n += pow(2, 32);
        }
        if (long_m < 0) {
            long_m += pow(2, 32);
        }
        
        // mask = 11111...111
        long long mask = pow(2, 32) - 1;
        int idx = i; // 2
        while (idx <= j) {
            mask -= pow(2, idx); // mask - 2^2 - 2^3... - 2^6
            idx++;
        }
        
        long long ans = mask & long_n; // ans = 10000000000
        ans += long_m * pow(2, i); // ans + 10101 << 2
        
        return (int)(ans > (long long)pow(2, 32) ? ans - (long long)pow(2, 32): ans);
    }
};