如何结合列表的take与drop函数，返回包含取删n元素结果的元组？

要实现一个同时返回take和drop结果的函数，有两种常见实现方式：一种是直接复用现有函数，另一种是写单次遍历的高效版本。

1. 简易复用版 #

如果不追求极致效率，直接把两个函数的结果打包成元组即可：

let take_drop_naive n l = (take n l, drop n l)

这种写法简单直观，但会两次遍历输入列表，列表较大时效率偏低。

2. 单次遍历高效版 #

更优的方式是写一个递归函数，一次遍历就能生成两个结果，还能处理原take/drop未覆盖的边界情况（比如n大于列表长度）：

let rec take_drop n l =
  if n = 0 then
    ([], l)
  else
    match l with
    | [] -> ([], [])  (* 处理n超过列表长度的场景 *)
    | h::t ->
        let (taken, dropped) = take_drop (n - 1) t in
        (h :: taken, dropped)

逻辑说明： #

• 当n = 0时，无需取任何元素，直接返回空列表和原列表；
• 当输入列表为空时，不管n值大小，返回两个空列表（避免原take函数在n>0且列表为空时触发模式匹配错误）；
• 列表非空时，递归处理剩余子列表t，把当前元素h加到递归得到的taken列表头部，dropped直接复用递归结果。

测试示例 #

调用take_drop 3 [1;2;3;4;5]会返回([1;2;3], [4;5])；调用take_drop 6 [1;2;3]会返回([1;2;3], [])。

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