如何将列表拆分为指定长度分组？Haskell函数实现报错求助

需求说明 #

需要实现一个列表分组函数，满足：

• 将输入列表拆分为每个包含n个元素的子列表，示例如下：

  n = 2
  [1, 2, 3, 4, 5, 6] -> [[1, 2], [3, 4], [5,6]]
  
  n = 3
  [1, 2, 3, 4, 5, 6] -> [[1, 2, 3], [4, 5, 6]]
  

• 边界规则：当n为0或大于列表长度时返回Left n；当n有效时返回拆分后的列表（Right [[a]]）

原代码问题分析 #

你编写的代码存在多处错误：

split :: Int -> [a] -> Either Int [[a]]
split n [a]
   |n <= length [a] = Right n (take n [a]) : (split n (drop n [a]))
   |n == 0 = Left n
   |otherwise = Left n

1. 模式匹配错误 #

split n [a]仅能匹配长度为1的列表，无法处理任意长度的输入列表，应该用split n xs来匹配所有列表情况。

2. Either构造器使用错误 #

Right构造器只接受一个参数，你写成Right n (take n [a])属于语法错误，正确的用法是Right [take n xs]，但递归时还需要处理结果拼接的类型匹配问题。

3. 递归逻辑类型不匹配 #

你试图用列表操作符:将Right值和递归调用结果拼接，但split返回的是Either Int [[a]]类型，不是列表类型，这种操作完全不符合Haskell的类型规则。

4. 边界条件顺序错误 #

应该优先判断n=0的情况，否则当n=0时会先进入n <= length [a]的分支，导致逻辑错误。

修正后的代码 #

版本1：显式递归处理Either分支 #

split :: Int -> [a] -> Either Int [[a]]
split n xs
  | n == 0 = Left n
  | n > length xs = Left n
  | null xs = Right []
  | otherwise = case split n (drop n xs) of
                  Right rest -> Right (take n xs : rest)
                  Left err -> Left err

版本2：用辅助函数简化逻辑 #

split :: Int -> [a] -> Either Int [[a]]
split 0 _ = Left 0
split n xs
  | n > length xs = Left n
  | otherwise = Right $ go xs
  where
    go [] = []
    go ys = take n ys : go (drop n ys)

代码说明 #

• 优先处理n=0的非法情况，直接返回Left 0
• 判断n是否大于列表长度，若是则返回Left n
• 合法情况下，用辅助函数go完成递归分组：空列表返回空分组，否则取前n个元素，再递归处理剩余部分，最后用Right包裹结果

内容的提问来源于stack exchange，提问作者Io12