具有模式匹配的递归列表处理

这里我们演示如何使用 OCaml 的模式匹配语法递归处理列表。

let rec map f lst =
  match lst with
  | [] -> []
  | hd::tl -> (f hd)::(map f tl)

在这种情况下,模式 [] 匹配空列表,而 hd::tl 匹配任何具有至少一个元素的列表,并将列表的第一个元素分配给 hd,列表的其余部分(可以为空)分配给 tl

请注意,hd::tl 是一种非常通用的模式,它将匹配任何非空的列表。我们还可以编写与具有特定数量元素的列表匹配的模式:

(* Return the last element of a list. Fails if the list is empty. *)
let rec last lst =
  match lst with
  | [] -> failwith "Empty list"
  | [x] -> x (* Equivalent to x::[], [x] matches a list with only one element *)
  | hd::tl -> last tl

(* The second to last element of a list. *)
let rec second_to_last lst =
  match lst with
  | [] -> failwith "Empty list"
  | x::[] -> failwith "Singleton list"
  | fst::snd::[] -> snd
  | hd::tl -> second_to_last tl

此外,OCaml 支持列表本身元素的模式匹配。我们可以更具体地了解列表中元素的结构,OCaml 将推断出正确的函数类型:

(* Assuming a list of tuples, return a list with first element of each tuple. *)
let rec first_elements lst =
  match lst with
  | [] -> []
  | (a, b)::tl -> a::(first_elements tl)
(* val first_elements : ('a * 'b) list -> 'a list = <fun> *)

通过将这些模式组合在一起,我们可以处理任何任意复杂的列表。