[英]Produce the infinite list [0, 1, -1, 2, -2, … in Haskell
因此,假设我们想在Haskell中生成列表[0, 1, -1, 2, -2, ...
实现这一目标最优雅的方式是什么?
我想出了这个解决方案:
solution = [0] ++ foldr (\(a,b) c->a:b:c) [] zip [1..] $ map negate [1..]
但我相信必须有更好的方法。
这似乎是理解的目的:
solution = 0 : [y | x <- [1..], y <- [x, -x]]
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
或者我们可以使用fromEnum
将布尔值转换为整数,如@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])
怎么样
concat (zipWith (\x y -> [x, y]) [0, -1 ..] [1 ..])
要么
concat (transpose [[0, -1 ..], [1 ..]])
?
怎么样:
tail $ [0..] >>= \x -> [x, -x]
在片刻的反思中,在我看来,使用nub
而不是tail
会更优雅。
另一种原始解决方
alt = 0 : go 1
where go n = n : -n : go (n+1)
你也可以在这里使用concatMap
而不是foldr
,并用[0, -1..]
替换map negate [1..]
[0, -1..]
:
solution = concatMap (\(a, b) -> [a, b]) $ zip [0, -1..] [1..]
如果你想使用negate
,那么这是另一种选择:
solution = concatMap (\(a, b) -> [a, b]) $ (zip . map negate) [0, 1..] [1..]
只是因为没人说:
0 : concatMap (\x -> [x,-x]) [1..]
迟到了,但这也会做到
solution = [ (1 - 2 * (n `mod` 2)) * (n `div` 2) | n <- [1 .. ] ]
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.