Haskell slice函数如何在不计算undefined输入时返回[]？

我正在完成Haskell作业，需要实现一个slice函数，类型为slice :: Int -> Int -> [a] -> [a]，功能是从列表xs中先删除a个元素，再取接下来的b个元素。

遇到的问题 #

我的实现无法通过测试用例slice 5 5 undefined，该用例要求返回[]，但实际运行会抛出异常：

*** Exception: Prelude.undefined
CallStack (from HasCallStack):
  undefined, called at HW2_Test.hs:78:20 in main:Main

当前实现代码 #

{-# LANGUAGE LambdaCase #-}
import Prelude hiding (take, drop)

take :: Int -> [a] -> [a]
take n xs 
  | n <= 0 = []
  | otherwise = case xs of
      [] -> []
      x : xs' -> x : take (n - 1) xs'

drop :: Int -> [a] -> [a]
drop n xs
  | n <= 0 = xs
  | otherwise = case xs of
      [] -> []
      _ : xs' -> drop (n - 1) xs'

slice :: Int -> Int -> [a] -> [a]
slice a b xs
  | b <= 0 = []
  | a < 0 = slice (a + 1) b xs
  | otherwise = take b (drop a xs)

已尝试的方法 #

• 通过先检查计数再模式匹配，让take和drop函数尽可能惰性
• 调整slice的守卫条件顺序，优先处理无需访问列表的情况
• 尝试多种守卫条件组合，避免触发列表求值

核心问题分析 #

调用slice 5 5 undefined时，代码会求值为take 5 (drop 5 undefined)。即使是惰性实现，drop 5 undefined也需要对列表进行模式匹配来跳过5个元素，这会强制求值undefined，最终抛出异常。

需求与背景 #

• 这是作业任务，要求slice 5 5 undefined :: [Int]必须返回[]
• 不能使用异常处理模块（Control.Exception）
• 已经实现了take 0 undefined返回[]的正确行为

请问有没有办法实现该行为？还是我对函数的要求存在误解？

解答 #

在标准Haskell的纯函数语义下，无法实现slice 5 5 undefined返回[]，原因如下：

• slice 5 5 undefined的语义是“跳过5个元素后取5个元素”，要完成这个操作必须访问列表的前5个元素，但undefined是一个bottom值（非终止表达式），任何对它的模式匹配都会触发异常。
• 纯Haskell函数无法区分undefined和一个真实的列表（比如无限列表），因此无法在不访问列表的情况下判断“跳过5个元素后是否还有元素可取”。

如果测试用例确实要求该行为，大概率是对函数语义的误解或测试用例存在笔误：

1. 若测试用例应为slice 5 0 undefined：你的现有实现已经能正确返回[]，因为b<=0的守卫会直接触发返回。
2. 若作业要求的是“当跳过a个元素后剩余列表长度不足b时返回[]”：对于正常列表这个逻辑是合理的，但对于undefined这类bottom值，无法在不触发异常的情况下判断剩余长度。

符合常规语义的优化实现 #

如果你需要的是符合标准slice语义（同时尽可能保持惰性）的实现，可以将drop和take的逻辑合并，减少不必要的列表遍历：

slice :: Int -> Int -> [a] -> [a]
slice a b xs = go (max 0 a) (max 0 b) xs
  where
    go skip takeCount [] = []
    go skip takeCount (x:xs')
      | takeCount <= 0 = []
      | skip > 0 = go (skip - 1) takeCount xs'
      | otherwise = x : go skip (takeCount - 1) xs'

这个实现会优先处理无需访问列表的情况，并且在遍历过程中同时跟踪跳过和取数的计数，比take b . drop a的组合更高效，但依然无法满足slice 5 5 undefined返回[]的要求——这在纯Haskell中是不可能完成的。

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