位运算(LeetCode基础)

位运算

位运算

• 按位与 &
• 按位或 |
• 按位异或 ^
• 左移 >>
• 右移 <<
• 取反 ~

应用

二进制枚举子集

1. 对于一个元素个数为n的集合S 来说，每一个位置上的元素都有选取和未选取两种状态。我们可以用数字1来表示选取该元素，用数字0来表示不选取该元素。
2. 那么我们就可以用一个长度为 $n$ 的二进制数来表示集合 $S$ 或者表示 $S$ 的子集。其中二进制的每一个二进位都对应了集合中某一个元素的选取状态。对于集合中第 $i$ 个元素来说，二进制对应位置上的 $1$ 代表该元素被选取，$0$ 代表该元素未被选取。

def subSet(S):
    n = len(S)
    sub_sets = []
    for i in range(1 << n):
        sub_set = []
        for j in range(n):
            if i >> j & 1:
                sub_set.append(S[j])
            sub_sets.append(sub_set)
        return sub_sets

练习

01颠倒二进制位

对于一个二进制数，我们可以用位运算来提取它每一位上的数字，然后将其反转后再合并成一个新的二进制数。

#颠倒二进制位
def reverse_bits(n: int) -> int:
    result = 0
    for i in range(32):
        result <<= 1
        result |= n & 1
        n >>= 1
    return result

02位1的个数

可以使用位运算的方法来统计二进制数中1的个数。一个简单的方法是使用循环和位运算的方式。

#统计1的个数
def count_ones(n: int) -> int:
    count = 0
    while n != 0:
        if n & 1 == 1:
            count += 1
        n >>= 1
    return count

03数字范围按位与

1. 我们需要找到区间 $[left,right]$中所有数字的共同前缀，然后将共同前缀的后面部分都置为0
2. 使用一个循环，不断将 left和right右移，直到 left 和 right 相等或其中一个为0
3. 在每次循环中，将 result 左移一位，并将最低位设为0

#数字范围按位与
def range_bitwise_and(left: int, right: int) -> int:
    result = 0
    while left != right and left != 0:
        left >>= 1
        right >>= 1
        result += 1
    result <<= 1
    return result