中等 更新二進位制位 (lintcode)
阿新 • • 發佈:2019-01-30
更新二進位制位
給出兩個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
最少的操作次數是多少?
- 如果j<31,即j不在最高位上。可以把i到j位清為0,可以
((1<<(j+1))-(1<<i))
得到i到j之間全是1的數,再取反,得到i到j之間全是0的數。 - 如果j=32,
(1<<(j+1))即(1<<32),相當於1<<1
(1<<i)-1
得到i到j之間全是0,其他地方是1的數。 - 上面得到的數成為掩碼
- (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); } };