Haskell 群函数/第一个元素总是丢失

Question

我想编写一个名为 group 的函数，它接受 [a] 类型的列表并返回包含相同字符的 [[a]] 类型列表的列表。以下是预期行为的简单示例:

组 [1,1,2,1,3,3,3] ⇒+ [[1,1],[2],[1],[3,3 ,3]]

组“aaabbcdddd” ⇒+ [“aaa”,“bb”,“c”,“dddd”]

我编写了以下代码，但问题是该代码总是错过原始列表中的第一个元素

group [] = []
group (x:xs) = groupHelper xs x []

groupHelper :: Eq a => [a] -> a -> [[a]] -> [[a]]
groupHelper [] prev fs = fs
groupHelper (x:xs) prev [] = groupHelper xs prev [[x]]
groupHelper (x:xs) prev fs = if (x == prev) then groupHelper (xs) (x) ((init fs) ++ [(last(fs)++[x])]) 
     else groupHelper (xs) x (fs++[[x]])

代码将为此输入生成以下输出。

组 [1,1,2,1,3,3,3] ⇒+ [[1],[2],[1],[3,3,3 ]]

=> 这里第一个列表开头的 1 会丢失

<小时/>

组“aaabbcdddd” ⇒+ [“aa”,“bb”,“c”,“dddd”]

=> 这里第一个列表开头的一个 'a' 将丢失

希望有人能帮我解决这个问题。非常感谢

Answer 1

当您第一次(从组)调用 groupHelper 时，您将传递 prev 参数，但助手仅将其用于比较。解决方案可能是在第一次调用时将第一个项目添加为第一组的第一个元素。像这样:group (x:xs) = groupHelper xs x [[x]]