[英]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.