簡體   English   中英

Scala-將聯接值映射到鍵?

[英]Scala - Map join values to keys?

我具有目標節點“ E”的以下映射:

val map = Map("A" -> "B", "A" -> "C", "C" -> "D", "C" -> "E")

它描述了一個有向節點圖,如下所示:

A
 / \
 B  C
   / \     
  D   E

我需要在任何時候輸入圖並生成到目標節點的路由。

Example 1: Enter at A -> Route: A->C->E
Example 2: Enter at D -> Route: D->C->E
Example 3: Enter at B -> Route: B->A->C->E

有誰知道一個緊湊的算法可以做到這一點,因為以前必須嘗試過。

期待您的來信。

干杯,

耶茲

就是這樣:

  val map = List("A" -> "B", "A" -> "C", "C" -> "D", "C" -> "E")

  def pathOf(tree: Iterable[(String,String)],from: String,to: String, path: List[String] = Nil): List[String] = {
    if(from == to) return to::path
    tree.filterNot{ case(a,b) => path.contains(a)||path.contains(b) }
    .collect{
      case (a,b) if a == to => b
      case (a,b) if b == to => a
    }.map{ x => pathOf(tree,from,x,to::path) }
    .find{ _.nonEmpty }
    .getOrElse(Nil)
  }

用例:

scala>  pathOf(map,"B","E").mkString("->")
res1: String = B->A->C->E

scala>  pathOf(map,"C","E").mkString("->")
res2: String = C->E

作為Scala的新手,我認為這個問題對我自己是一個很好的練習,並希望與大家分享我的解決方案。 歡迎任何評論!

順便說一句,@ Eastsun提供的解決方案是深度優先搜索,它“存儲”每個路徑中的訪問節點,而我的是廣度優先搜索,其中不需要存儲(盡管您可以添加此功能以提高效率)。 對於樹木,它們給出相同的答案,但是對於一般圖形,它們可以不同。

每個節點的鄰居也可以緩存以進行優化。

val graph = Vector(("A","B"), ("A","C"), ("C","D"), ("C","E"))
def adjacent(a: String) = {
  graph flatMap {
    case (`a`, x) => Some(x)
    case (x, `a`) => Some(x)
    case _ => None
  }
}
def go(from: String, to: String) {
  def expand(paths: Vector[Vector[String]]) {
    paths.find(_.last==to) match {
      case Some(x) => println(x); return
      case None => expand(paths flatMap { e => 
        adjacent(e.last) map (e :+ _) 
      })
    }
  }
  expand(Vector(Vector(from)))
}
// tests
go("A","E")  // Vector(A, C, E)
go("B","E")  // Vector(B, A, C, E)
go("D","E")  // Vector(D, C, E)

帶記憶的版本:更改

adjacent(e.last) map (e :+ _)

adjacent(e.last) filterNot (x => paths.flatten contains x) map (e :+ _)

或將此功能放在相鄰功能中。

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM