Leetcode-位运算

191. 位1的个数 https://leetcode-cn.com/problems/number-of-1-bits/

编写一个函数,输入是一个无符号整数,返回其二进制表达式中数字位数为 ‘1’ 的个数(也被称为汉明重量)。

提示:

请注意,在某些语言(如 Java)中,没有无符号整数类型。在这种情况下,输入和输出都将被指定为有符号整数类型,并且不应影响您的实现,因为无论整数是有符号的还是无符号的,其内部的二进制表示形式都是相同的。
在 Java 中,编译器使用二进制补码记法来表示有符号整数。因此,在上面的 示例 3 中,输入表示有符号整数 -3。

解:

每次mod 2以后再右移一位,枚举所有位数。

class Solution(object):
    def hammingWeight(self, n):
        """
        :type n: int
        :rtype: int
        """
        res = 0
        mask = 1
        for i in range(32):
            if n & mask:
                res += 1
            mask <<= 1
        return res

  

x & (x-1),例如1100100 & 1100011 结果为1100000,把x最后一位的1给消除了,因为减1的过程中需要向最低位的1借位,所以与运算后最低位1前面的数都不变。while(x != 0): count++; x=x&(x-1)。

class Solution(object):
    def hammingWeight(self, n):
        """
        :type n: int
        :rtype: int
        """
        res = 0
        mask = n
        while mask:
            mask &= mask - 1
            res += 1
        return res

  

 

231. 2的幂 https://leetcode-cn.com/problems/power-of-two/

给定一个整数,编写一个函数来判断它是否是 2 的幂次方。

解:

不断的mod 2

class Solution(object):
    def isPowerOfTwo(self, n):
        """
        :type n: int
        :rtype: bool
        """
        if n <= 0:
            return False
        
        while n > 1:   # 如果一直能被2整除,直到最后等于1,即为2的幂次
            if n % 2 == 0:
                n //= 2
            else:
                return False
        return True

  

位运算 x=x&(x-1),所有2的幂次都满足有且仅有最高位一个1。直接判断 x != 0 且 x & (x-1) == 0

class Solution(object):
    def isPowerOfTwo(self, n):
        """
        :type n: int
        :rtype: bool
        """
        if n <= 0:
            return False
        
        return n&(n-1) == 0

  

338. 比特位计数 https://leetcode-cn.com/problems/counting-bits/

给定一个非负整数 num。对于 0 ≤ i ≤ num 范围中的每个数字 i ,计算其二进制数中的 1 的数目并将它们作为数组返回。

进阶:

给出时间复杂度为O(n*sizeof(integer))的解答非常容易。但你可以在线性时间O(n)内用一趟扫描做到吗?
要求算法的空间复杂度为O(n)。
你能进一步完善解法吗?要求在C++或任何其他语言中不使用任何内置函数(如 C++ 中的 __builtin_popcount)来执行此操作。

解:

借助#191统计一个数中1的个数,外面再写一个循环,O(N*32)

class Solution:
    def countBits(self, num: int) -> List[int]:
        if num <= 0:
            return [0]
        res = [0]
        for i in range(1, num+1):
            res.append(self.hammingWeight(i))
        return res
    
    def hammingWeight(self, n):
        res = 0
        mask = n
        while mask:
            mask &= mask - 1
            res += 1
        return res

  

动态规划,最后设置位。count[i] = count[ i & (i-1) ] + 1。i中1的个数,等于i&(i-1)中1的个数再加1

class Solution:
    def countBits(self, num: int) -> List[int]:
        res = [0] * (num + 1)
        for i in range(1, num+1):
            res[i] = res[i & (i-1)] + 1
        return res

  

动态规划,最低有效位。一个数i中1的个数,等于 i mod 2的结果 + i//2中1的个数

class Solution:
    def countBits(self, num: int) -> List[int]:
        res = [0] * (num + 1)
        for i in range(1, num+1):
            res[i] = res[i//2] + i%2
        return res

  

动态规划,最高有效位。count[i+b] = count[i] + 1,其中b取第一个大于i的2的幂次。例如 i=101,b=1000

class Solution:
    def countBits(self, num: int) -> List[int]:
        res = [0] * (num + 1)
        
        i, b = 0, 1
        while b <= num:
            while i < b and i + b <= num:   # 把小于当前b的i全部计算完,再更新b
                res[i+b] = res[i] + 1
                i += 1
            i = 0   # reset i 
            b <<= 1   # b = 2b
        return res

  

52. N皇后ii  https://leetcode-cn.com/problems/n-queens-ii/

皇后问题研究的是如何将 n 个皇后放置在 n×n 的棋盘上,并且使皇后彼此之间不能相互攻击。

给定一个整数 n,返回 n 皇后不同的解决方案的数量。

解:

另一种解法,位运算,使用bitmap回溯。用整型的二进制位来表示能放皇后的位置和不能放的位置。

class Solution:
    def totalNQueens(self, n: int) -> int:
        if n <= 0:
            return [] 
        
        count = 0
        
        def dfs(row=0, cols=0, hills=0, dales=0):
            """
            row: 当前放置皇后的行号
            cols: 列占据情况 [1 = 被占据,0 = 未被占据]
            hills: 主对角线占据情况 [1 = 被占据,0 = 未被占据]
            dales: 次对角线占据情况 [1 = 被占据,0 = 未被占据]
            """
            nonlocal count, n
            if row == n:  # 停止条件,如果已经放置了 n 个皇后
                count += 1  # 累加可行解,这轮结束
                return
            # 列、两对角线先或,得到所有可能空位。取反后和1左移n位-1的结果(筛子)相与,保留后9位的同时把高位的1全部清掉,剩下为1的位置在这一行可以放皇后
            # 例如 1 0111 0110,取反0 1000 1001,1左移9位-1后01 1111 1111,结果为 0 1000 1001,为1的位表示可以放置的列
            bitmap = (~(cols|hills|dales)) & ((1 << n) - 1)
            
            while bitmap:
                p = bitmap & (-bitmap)   # 取到最低位的1。取反再加1,比最后一个1更高的位都不同,其他位都相同,相与的话取到最低位的1
                dfs(row + 1, cols|p, (hills|p)>>1, (dales|p)<<1)  # 下一行。列中,当前放置的p被占,主对角线col+1被占,次对角线col-1被占。对角线初始和列相同,之后更新根据主次右左移
                bitmap = bitmap & (bitmap - 1)  # 去掉最低位的1
                
        dfs()
        return count

  

posted @ 2019-08-19 10:28  王朝君BITer  阅读(268)  评论(0编辑  收藏  举报