我想要一个带有位掩码Int
的函数,并将其掩码值作为一组Int
返回。像这样:
func split(bitmask: Int) -> Set<Int> {
// Do magic
}
这样
split(bitmask: 0b01001110) == [0b1000000, 0b1000, 0b100, 0b10]
答案 0 :(得分:3)
一种解决方案是检查每个位并在该位置位时添加相应的掩码。
func split(bitmask: Int) -> Set<Int> {
var results = Set<Int>()
// Change 31 to 63 or some other appropriate number based on how big your numbers can be
for shift in 0...31 {
let mask = 1 << shift
if bitmask & mask != 0 {
results.insert(mask)
}
}
return results
}
print(split(bitmask: 0b01001110))
对于二进制数0b01001110
,结果将为:
[64,2,4,8]
这是你问题中结果的十进制等值。
对于十六进制数0x01001110
(二进制为1000100010000
),结果将为:
[16,256,4096,16777216]
这是另一种解决方案,不需要知道值的大小,对于较小的数字,它的效率稍高:
func split(bitmask: Int) -> Set<Int> {
var results = Set<Int>()
var value = bitmask
var mask = 1
while value > 0 {
if value % 2 == 1 {
results.insert(mask)
}
value /= 2
mask = mask &* 2
}
return results
}
答案 1 :(得分:3)
请注意,位掩码的最常见用例包括将特定的,有意义的布尔标志的集合打包到单个字大小的值中,并对这些标志执行测试。 Swift在OptionSet
类型中为此提供了便利。
struct Bits: OptionSet {
let rawValue: UInt // unsigned is usually best for bitfield math
init(rawValue: UInt) { self.rawValue = rawValue }
static let one = Bits(rawValue: 0b1)
static let two = Bits(rawValue: 0b10)
static let four = Bits(rawValue: 0b100)
static let eight = Bits(rawValue: 0b1000)
}
let someBits = Bits(rawValue: 13)
// the following all return true:
someBits.contains(.four)
someBits.isDisjoint(with: .two)
someBits == [.one, .four, .eight]
someBits == [.four, .four, .eight, .one] // set algebra: order/duplicates moot
someBits == Bits(rawValue: 0b1011)
(在实际使用中,您当然会在OptionSet
类型中为每个&#34;元素&#34;值提供一些对您的使用有意义的值情况)。
OptionSet
实际上是一个单独的值(根据自身而不是元素类型支持设置代数),所以它不是一个集合 - 也就是说,它不是&# 39;提供一种枚举其元素的方法。但是,如果您打算使用位掩码的方式只需要设置和测试特定的标志(或标志组合),那么您可能不需要一种枚举元素的方法。
如果您确实需要枚举元素,但又需要OptionSet
的所有设置代数功能,则可以将OptionSet
与位分割数学结合起来,例如在@rmaddy's answer中找到的数学:
extension OptionSet where RawValue == UInt { // try being more generic?
var discreteElements: [Self] {
var result = [Self]()
var bitmask = self.rawValue
var element = RawValue(1)
while bitmask > 0 && element < ~RawValue.allZeros {
if bitmask & 0b1 == 1 {
result.append(Self(rawValue: element))
}
bitmask >>= 1
element <<= 1
}
return result
}
}
someBits.discreteElements.map({$0.rawValue}) // => [1, 4, 8]
答案 2 :(得分:1)
这是我的&#34; 1行&#34;版本:
func split(bitmask: Int) -> Set<Int> {
return Set(Array(String(bitmask, radix: 2).characters).reversed().enumerated().map { (offset, element) -> Int in
Int(String(element))! << offset
}.filter { $0 != 0 })
}
效率不高,但很有趣!
编辑:包含在拆分功能...
let values = Set(String(0x01001110, radix: 2).utf8.reversed().enumerated().map { (offset, element) -> Int in
Int(element-48) << offset
}.filter { $0 != 0 })
编辑稍微短暂
{{1}}