你如何在Go中设置和清除一个位?

时间:2014-04-21 06:42:29

标签: go bit-manipulation

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

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

2 个答案:

答案 0 :(得分:29)

这是设置位的功能。首先,将数字1移动整数中指定的空格数(因此它变为0010,0100等)。然后用原始输入OR。这使得其他位不受影响,但始终将目标位设置为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)。然后使用按位AND,它不会触及数字AND&1,但会取消设置为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等),然后将其与目标数字进行AND运算。如果结果数大于0(它将是1,2,4,8等),则设置该位。

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

答案 1 :(得分:14)

还有一个紧凑的符号可以清除一点。该运算符为&^,称为“而不是”。

使用此运算符,clearBit函数可以这样写:

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

或者像这样:

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