Question

我有一个由0＆＃39; s和1组成的一维Repa数组，我想计算它的游程编码。例如：转动[0,0,1,1,1,0,0,0,1,0,1,1] into [2,3,3,1,1,2]或类似的东西。（由于可读性，我使用列表表示）

理想情况下，我想要1的游程并忽略0。所以[0,0,1,1,1,0,0,0,1,0,1,1] becomes [3,1,2]。

我希望结果也是（Repa）数组。

如何使用Repa执行此操作？我无法使用map或traverse，因为他们一次只能给我一个元素。我可以试着fold使用一些特殊的累加器，但这似乎并不理想，我甚至不知道它（由于monad定律）。

Answer 1

我目前正在迭代数组并返回列表而不使用任何Repa函数。我正在研究Boolean而不是1和0，但算法是相同的。我之后将此列表转换为Repa数组。

runLength :: Array U DIM1 Bool -> [Length]
runLength arr = go ([], 0, False) 0 arr
  where
    Z :. n = extent arr
    go :: Accumulator -> Int -> Array U DIM1 Bool -> [Length]
    go !acc@(xs, c, b) !i !arr | i == n = if c > 0 then c:xs else xs
                               | otherwise =
                                 if unsafeIndex arr (Z :. i)
                                 then if b
                                      then go (xs, c+1, b) (i+1) arr
                                      else go (xs, 1, True) (i+1) arr
                                 else if b
                                      then go (c:xs, 0, False) (i+1) arr
                                      else go (xs, 0, False) (i+1) arr

Repa数组的行程编码

1 个答案: