Haskell如何删除列表的所有0,直到其中一个元素/=0?

我是 Haskell 的新手,我正在尝试弄清楚如何编写代码,该代码将删除所有零,直到列表中的元素 > 0。

例如:

输入:[0,0,5,6,0]

输出:[5,6,0]

到目前为止,我已经写了这个:

zeroUntil :: [Int] -> [Int]
zeroUntil [] = []
zeroUntil (x:xs)
    | x == 0 = drop x (xs)
    | otherwise = zeroUntil xs

但不知何故[5,6,0]我得到[]

任何人都可以解释我做错了什么吗?

回答

您可以使用dropWhile :: (a -> Bool) -> [a] -> [a]并丢弃物品,只要这些物品为零:

zeroUntil :: [Int] -> [Int]
zeroUntil = dropWhile (0 ==)

如果您想删除零直到元素大于零,您可以使用递归。如果这些小于零,您的递归案例应该产生元素:

zeroUntil :: [Int] -> [Int]
zeroUntil [] = []
zeroUntil (0:xs) = zeroUntil xs
zeroUntil (x:xs)
    | x > 0 = x : xs  -- ← end of recursion, return the list
    | otherwise = x : zeroUntil xs  -- ← yield x and recurse

例如:

Prelude> zeroUntil [0,0,5,6,0]
[5,6,0]
Prelude> zeroUntil [0,-1,5,6,0]
[-1,5,6,0]


以上是Haskell如何删除列表的所有0,直到其中一个元素/=0?的全部内容。
THE END
分享
二维码
< <上一篇
下一篇>>