除非性能很关键,否则我建议使用位矢量表示法来表示这样的项目。正如你发现的那样,随机访问单个位在打包形式时是一件很痛苦的事情,但Data.Vector
为这类任务提供了丰富的功能。
import Data.Bits
import qualified Data.Vector as V
type BitVector = V.Vector Bool
unpack :: (Bits a) => a -> BitVector
unpack w = V.generate (bitSize w) (testBit w)
pack :: (Bits a) => BitVector -> a
pack v = V.ifoldl' set 0 v
where
set w i True = w `setBit` i
set w _ _ = w
mkPermutationVector :: Int -> V.Vector Int
mkPermutationVector d = V.generate (2^d) b
where
b i | i < 2^(d-1) = 2*i
| otherwise = let i' = i-2^(d-1)
in 2*i'+1
permute :: Int -> BitVector -> BitVector
permute d v = V.backpermute v (mkPermutationVector d)
请注意,这是如何让你通过密切转录数学描述来指定排列。这大大减少了错误发生的可能性,并且比编写代码的代码更令人愉快。
当您例如向量(以10为基数)测试:现在
*Main> import Data.Word
*Main Data.Word> let permute16 = pack . permute 4 . unpack :: Word16 -> Word16
*Main Data.Word> permute16 43690
65280
,通过移动到位向量作为你的表现,你使用的Haskell类型,这样会失去很多,你得到了什么免费作为Num
实例。但是,您始终可以为您的表示实施Num
操作;这里是一个开始:
plus :: BitVector -> BitVector -> BitVector
plus as bs = V.tail sums
where
(sums, carries) = V.unzip sumsAndCarries
sumsAndCarries = V.scanl' fullAdd (False, False) (V.zip as bs)
fullAdd (_, cin) (a, b) = ((a /= b) /= cin
, (a && b) || (cin && (a /= b)))
您也可以找到勒Erkok的sbv
包是有用的,但我不知道它暴露一样便利backpermute
功能为您的特定问题。
更新:我认为这是一个有趣的问题来回答,所以我继续充实码出位的库:bit-vector。
http://haskell.org/ghc/docs/latest/html/libraries/base/Data-Bits.html#t:Bits – 2011-09-04 10:33:39