haskell - 基于发现值的映射

Question

我希望能够根据找到的元素映射但跳过一些元素。问题的本质就在这里。

a = [1,2,3,4,5]

i = 0
while i < len(a):
    if a[i] % 2 == 0:
       a[i] *= 2
    else:
       i += 1
    i += 1

我可以通过传递一个索引变量和一个累加器在 Haskell 中用显式递归来表达这一点。

f :: Int -> [Int] -> [Int] -> [Int]
f index list acc = if index < length list then 
                   if even (list !! index) then f (index + 1) list (acc ++ [(list !! index) * 2]) else f (index + 2) list (acc ++ [list !! index, list !! (index + 1)])
                   else acc

f 0 [1,2,3,4,5] []

但这感觉真的很难看，有没有更好的方法呢？

Answer 1

如何使用 mapAccumL 并使用 bool 累加器来表示前一个数字是否为奇数？

f = snd . mapAccumL go False
  where
    go False n | even n = (False, n * 2)
    go True n | even n = (False, n)
    go _ n = (True, n)