Кодировка длины строки массива Repa

Question

У меня есть одномерный массив Repa, который состоит из 0 и 1, и я хочу рассчитать его кодировку длины пробега. Например: поворот [0,0,1,1,1,0,0,0,1,0,1,1] into [2,3,3,1,1,2] или что-то подобное. (Я использую представление списка из-за удобочитаемости)

В идеале, мне бы хотелось, чтобы длина пробега 1 и игнорировались 0. So [0,0,1,1,1,0,0,0,1,0,1,1] becomes [3,1,2].

Я хотел бы получить результат (Repa).

Как это сделать, используя Repa? Я не могу использовать map или traverse, так как они дают мне только один элемент за раз. Я мог бы попробовать fold с каким-то специальным аккумулятором, но это не кажется идеальным, и я не знаю, что это даже возможно (из-за законов монады).

Answer 1

Я сейчас просто перебираю массив и возвращаю список, не используя никакой функции Repa. Я работаю над Boolean s вместо 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