Ocaml：两个整数的函数并返回范围内所有整数的列表

Question

这个函数接受两个整数并返回[a,b]范围内所有整数的列表

这是我写的解决方案。

let rec range_rec l a b = 
  if (a=b) then l@[b]
  else range_rec (l@[a], a+1, b);;

let range a b = range_rec [] a b;;

我遇到了一个错误“错误：此表达式的类型为 int list * int * int 但表达式应为 int 类型”。有人能解释一下我为什么会收到这个错误吗？

谢谢。

Answer 1

为了改进 gasche 已经提出的解决方案，可以将其设为尾递归。

let int_range a b =
  let rec int_range_rec l a b =
    if a > b then l
    else int_range_rec (b :: l) a (b - 1)
  in (int_range_rec [] a b);;

Answer 2

应该是这样的：

let rec range_rec l a b = 
  if a = b then l @ [b]
  else range_rec (l @ [a]) (a + 1) b;;

let range a b = range_rec [] a b;;

我做了什么：

1. 将loop 更改为range_rec
2. 将(l@[a], a+1, b) 更改为(l @ [a]) (a + 1) b。第一个是三元组，第二个是柯里化函数的 3 个参数。
3. 注意if (a = b) then可以写成if a = b then。

最后，通过“向后”循环使用:: 而不是@，可以提高函数的效率。例如像gasche所展示的那样。

Answer 3

从性能的角度来看，l @ [elem] 操作很糟糕：因为a @ b 在a 的长度上是线性的，所以在列表末尾添加一个元素在列表的长度上是线性的，使你的整个@ 987654324@ 在|b-a| 中定义二次方。可行的方法是更改​​您的代码，以便您可以改用恒定时间elem::l 操作。

let rec range a b =
  if a > b then []
  else a :: range (a + 1) b

您可以进行额外的优化，例如使其尾递归，但至少该解决方案的复杂性是正确的。