scala:在具有可选返回类型的树上查找方法

Question

我正在树状结构上编写查找方法。这是我的代码:

def find[T](node: Node, findThisGuy: T): Option[Node] = {
    if (node.data == findThisGuy) {
        Some(node)
    } else {
        if (node.children.nonEmpty) {
            node.children.foreach(child => {
                find(child, findThisGuy)
            })
        } else {
            None
        }
    }
}

这里的问题是内部 if 具有 Unit 的返回类型，它确实符合函数的返回类型，即 Option[Node]。那么如何定义内部if的返回类型呢？我愿意就更好地实现上述方法提出建议。

Answer 1

我认为这会成功:

def find[T](node: Node, findThisGuy: T): Option[Node] = {
  if (node.data == findThisGuy) {
    Some(node)
  } else {
    node.children.view
      .map(child => find(child, findThisGuy))
      .collectFirst{case Some(guy) => guy}
  }
}