ocaml - 平方和的总和OCaml

我正在尝试通过使用foldl函数求和 float 列表中两个连续元素的所有平方和。

let rec foldl (f: 'b -> 'a -> 'b) (accum: 'b) (lst: 'a list) : 'b = match lst with
|[] -> accum
|x::xs -> foldl f (f accum x) xs

let sum_sqrt_sums (mylist:  float list) : float = match mylist with
 |[] -> raise(Failure "Nope")
 |[x] -> raise(Failure "No!")
 |x::xs -> foldl (fun x y -> sqrt (x +. y)) x xs

我跑步时有两个不同的结果

sum_sqrt_sums [4.0; 2.0; 6.0; 3.0];;
- : float = 2.43039103901312092

sqrt(4.0 +. 2.0) +. sqrt(2.0 +. 6.0) +. sqrt(6.0 +. 3.0) ;;
- : float = 8.27791686752936862

我的求和函数中的逻辑有什么问题？

最佳答案

您的函数sum_sqrt_sums无法计算

sqrt(4.0 +. 2.0) +. sqrt(2.0 +. 6.0) +. sqrt(6.0 +. 3.0)

但

sqrt (sqrt (sqrt(2.0 +. 4.0) +. 6.0) +. 3.0)

您要做的是保留累加器中的最后一个元素，将其添加到下一个元素，并将它们的平方和加到累加器中:

let sum_sqrt_sums = function
  | [] | [_] -> raise(Failure "Nope")
  | x::xs ->
     let _, res = foldl (fun (x, acc) y -> (y, sqrt (x +. y) +. acc)) (x, 0.) xs in
     res

(请注意，您的foldl函数是List.fold_left函数)

更新(使用不同变量名的版本，以避免混淆):

let sum_sqrt_sums = function
  | [] | [_] -> raise(Failure "Nope")
  | x::xs ->
     let _, res = foldl (fun (e, acc) y -> (y, sqrt (e +. y) +. acc)) (x, 0.) xs in
     res