Question

我有以下对象结构：

case class Node(id:Int,children:List[Node])

示例：

NodeA
    id: 1
    children:[
        NodeA1:
            id: 2
            children:
                NodeA11
                ...
        ]       
        NodeB1:
            id: 3
            children:[]
        NodeC1:
            id: 17
            children: [
                NodeC11:
                    id:11
                    children:[
                        NodeC111:
                            id: 19
                            children: []
                    ]

            ]
        ...

我想创建一个递归循环来获取具有特定Id的Node，但是如果找不到iD并且对象上有任何对象，我仍然坚持如何继续运行功能儿童名单。我的函数只能用于获取第一个节点（例如：Id = 1）。

以下是我要做的事情：

def getNode(id:Int, node:Node) : Node = {
    var result:Node = null
    if(node.id == id){
      return node
    } else if(node.children.size > 0 ){
      for(children <- node.children){
        result = getNode(id, children)
        if(result.id == id){
          return result
        }
      }
    }
    return result
}

Answer 1

函数getNode确实应该返回Option[Node]来解释id树中缺少Node的搜索。

在这种情况下，您可以编写递归调用的选项：

def getNode(id:Int, node:Node): Option[Node] = 
  if (node.id == id) Some(node)
  else node.children.collectFirst(Function.unlift(getNode(id, _)))

在命令性案例中，您不需要检查列表长度：只需在检查每个孩子的循环之后返回None / null（或者不检查是否没有儿童）。

def getNode(id:Int, node:Node) : Option[Node] = {
  if (node.id == id) Some(node)
  else {
    for (child <- node.children) {
      val result = getNode(id, child)
      // At this point `result` is Some(nodeWeAreLookingFor) 
      // if it is in the subtree of the current `child`
      // or None otherwise
      if (result.isDefined) return result
    }
    None
  }
}

对于Java，您当然可以用Option替换null，但在Scala中，这个想法自然地由Option

建模

Scala - 递归函数不返回对象

1 个答案: