Haskell自定义数组索引函数：如何检查并处理索引越界问题？

你的原代码存在两个问题：

• 当索引为负数时，drop n xs会返回原列表，head会直接取第一个元素，不会触发错误
• 当索引大于等于列表长度时，drop n xs得到空列表，head会抛出默认的Prelude.head: empty list错误，不是你需要的指定信息

这里提供两种实现方式满足你的需求：

方式一：直观的长度检查版本 #

这个版本逻辑简单，适合新手理解：

atIndex :: [a] -> Int -> a
atIndex xs n
  | n < 0 || n >= length xs = error "error, your index is out of range"
  | otherwise = head (drop n xs)

通过守卫语句先判断索引是否合法：小于0或者大于等于列表长度时，抛出指定错误；否则执行原有的取值逻辑。

方式二：递归优化版本 #

这个版本避免了length遍历整个列表的开销，更符合Haskell的递归风格：

atIndex :: [a] -> Int -> a
atIndex [] _ = error "error, your index is out of range"
atIndex (x:_) 0 = x
atIndex (_:xs) n
  | n < 0 = error "error, your index is out of range"
  | otherwise = atIndex xs (n - 1)

逻辑说明：

1. 空列表直接抛出指定错误（对应索引超过列表长度的情况）
2. 索引为0时，直接返回当前列表的第一个元素
3. 索引大于0时，递归处理剩余列表，索引减1
4. 索引小于0时，直接抛出指定错误

内容的提问来源于stack exchange，提问作者SS.h777