go - 你将如何在 Go 中设置和清除单个位？

Question

在 Golang 中，如何设置和清除整数的各个位？例如，行为如下的函数:

 clearBit(129, 7) // returns 1
 setBit(1, 7)     // returns 129

Answer 1

这是一个设置位的函数。首先，将数字 1 移动整数中指定的空格数(因此变为 0010、0100 等)。然后将其与原始输入进行或。这不会影响其他位，但将始终将目标位设置为 1。

// Sets the bit at pos in the integer n.
func setBit(n int, pos uint) int {
    n |= (1 << pos)
    return n
}

这是一个清除一点的函数。首先将数字 1 移动整数中指定的空格数(因此它变为 0010、0100 等)。然后使用 ^ 运算符翻转掩码中的每一位(因此 0010 变为 1101)。然后使用按位与，它不会触及带有 1 的数字 AND，但会取消设置掩码中设置为 0 的值。

// Clears the bit at pos in n.
func clearBit(n int, pos uint) int {
    mask := ^(1 << pos)
    n &= mask
    return n
}

最后，这是一个检查是否设置了位的函数。将数字 1 移动指定数量的空格(因此变为 0010、0100 等)，然后将其与目标数字相加。如果结果数大于 0(它将是 1、2、4、8 等)，则设置该位。

func hasBit(n int, pos uint) bool {
    val := n & (1 << pos)
    return (val > 0)
}