Haskell 相当于 Ruby 的 slice_when

Question

我是 Haskell 新手，我想要一个像 Ruby 的 slice_when 一样工作的函数:

slice_when ruby

a = [1,2,4,9,10,11,12,15,16,19,20,21]
b = a.slice_when {|i, j| i+1 != j }
p b.to_a #=> [[1, 2], [4], [9, 10, 11, 12], [15, 16], [19, 20, 21]]

haskell

ghci > slice_when (\x y -> x+1 /= y) [1,2,4,9,10,11,12,15,16,19,20,21] 

=> [[1, 2], [4], [9, 10, 11, 12], [15, 16], [19, 20, 21]]

非常感谢。

Answer 1

这是一个 sliceWhen 实现，我相信它可以像您想要的那样工作:

sliceWhen :: (a -> a -> Bool) -> [a] -> [[a]]
sliceWhen _ []  = []
sliceWhen _ [x] = [[x]]
sliceWhen f (x:y:xs)
  | f x y     = [x] : sliceWhen f (y:xs)
  | otherwise = let z:zs = sliceWhen f (y:xs)
                in (x:z) : zs

上面的函数在每个边界上运行提供的谓词，如果它返回True，则会创建一个接缝。否则，它将“当前”值附加到列表其余部分的第一个元素上。