Question

在OCaml中，如果我必须使用许多if-then-else编写函数，下面是我的愚蠢和丑陋的解决方案。

let foo () =
  let a1 = ... in
  if (a1) then
    result1
  else
    let a2 = ... in
    if (a2) then
      result2
    else
      let a3 = ... in
      if (a3) then
        result3
      else
        let a4 = ... in
        if (a4) then
          result4
        else
          result5.

如何美化上面的代码？我喜欢C / C ++＆amp;使用“return”来保存下一个if语句的缩进的Java样式。我可以用OCaml做同样的事情吗？

int foo () = {
  bool a1 = ...;
  if (a1)
    return result1;

  bool a2 = ...;
  if (a2)
    return result2;

  bool a3 = ...;
  if (a3)
    return result3;

  bool a4 = ...;
  if (a4)
    return result4;

  return result5;
}

Answer 1

OCaml中没有return语句，但您可以在例外的帮助下模拟一个语句：

exception Ret of t

let my_ret x = raise (Ret x)

let foo () =
 try
  let a1 = ... in
  if a1 then my_ret result1;
  let a2 = ... in
  if a2 then my_ret result2;
  ...
 with Ret x -> x

另一个有用的解决方案是使用延迟评估：

let foo () =
 let a1 = lazy ...
 and a2 = lazy ...
 and a3 = lazy ...
 in
 match a1, a2, a3 with
 | lazy true, _, _ -> result1
 | _, lazy true, _ -> result2
 | _, _, lazy true -> result3
 | _, _, _ -> result4

这是使用懒惰的例子之一，可能有更简洁的方式来表达你的计算。

Answer 2

核心库提供with_return函数，允许您从函数中执行非本地存在：

open Core_kernel.Std
let foo () = with_return (fun goto -> 
    if a1 then goto.return 1;
    if a2 then goto.return 2;
    if a3 then goto.return 3;
    if a4 then goto.return 4;
    if a5 then goto.return 5;
    return 6)

但通常最好使用模式匹配或重新考虑代码。例如，如果您有一个谓词列表，并且根据谓词是什么，您想要返回一个值，这意味着您可以将其编码为某个映射结构中的搜索：

let foo () = [
  clause1, expr1;
  clause2, expr2;
  clause3, expr3;
] |> List.Assoc.find true 
  |> Option.value ~default:expr4

当然，在这种情况下，您没有进行短路评估。您可以使用延迟评估或使用thunk来解决此问题。但除非你的计算非常重或产生副作用，否则它不值得。

Answer 3

if句法结构确实在 C 和 OCaml 中的工作方式不同。在 C 中，if语法形式是语句，在 OCaml 中它们是表达式。您在 C 中与 OCaml if最接近的是?:三元运算符。如果您尝试使用此运算符而不是if重写C代码，您将面临同样的挑战。但这并不意味着它是不可能的，因为其他答案会为您提供解决方案。

最简单的一种，它适用于两种语言，是在几个子函数^（*）中剪切函数体，并使用continuation：

let rec foo () =
  let a1 = … (* computation *) in
  if a1
  then result1
  else foo2 ()
and foo2 () =
  let a2 = … in
  if a2
  then result1
  else foo3 ()
and foo3 () = … (* etc *)

编写对象方法时可能仍然有点麻烦，但是你总是可以使用内部函数来重新获得缩进平衡＆＃34;在方法范围内。

另请注意，rec关键字的唯一目的是允许每个延续在源布局中跟随其调用者，此处没有真正的递归。

（*）：@ gsg还在评论中提到了它。

Answer 4

与if表达式不同，match子句扩展到函数的末尾，即使它们包含多个语句，也不需要括号。所以你可以这样做：

let foo () =
  match ... with
  | true -> result1
  | false ->
  match ... with
  | true -> result2
  | false ->
  match ... with
  | true -> result3
  | false ->
  match ... with
  | true -> result4
  | false -> result5

您没有在示例中显示result1来自哪里，所以我无法确定，但您可能会发现让...返回结果选项而不是布尔，例如

let foo () =
  match ... with
  | Some result1 -> result1
  | None ->
  ...

在OCaml中编写漂亮的多级嵌套if-then-else代码？

4 个答案: