在Haskell中生成无限列表[0,1,-1,2,-2,...

Question

所以假设我们想[0, 1, -1, 2, -2, ...在Haskell中生成列表.

实现这一目标最优雅的方式是什么？

我想出了这个解决方案:

solution = [0] ++ foldr (\(a,b) c->a:b:c) [] zip [1..] $ map negate [1..]

但我相信必须有更好的方法.

Answer 1

这似乎是理解的目的:

solution = 0 : [y | x <- [1..], y <- [x, -x]]

Answer 2

同 iterate

也许更优雅的方法是使用iterate :: (a -> a) -> a -> [a]每次生成下一个项目的函数.例如:

solution = iterate nxt 0
    where nxt i | i > 0 = -i
                | otherwise = 1-i

或者我们可以用if- then- 来内联这个else:

solution = iterate (\i -> if i > 0 then -i else 1-i) 0

或者我们可以将布尔值转换为整数,如@melpomene所说,用fromEnum,然后使用它来添加1或0回答,所以:

solution = iterate (\i -> fromEnum (i < 1)-i) 0

哪个更无瑕疵:

import Control.Monad(ap)

solution = iterate (ap subtract (fromEnum . (< 1))) 0

同 (<**>)

我们也可以使用<**>来自应用程序的运算符来生成每次正数和负数变量的数字,如:

import Control.Applicative((<**>))

solution = 0 : ([1..] <**> [id, negate])