我想链接2列唯一标识符,并且能够通过第二列值获取第一列值,并通过第一列值获取第二列值。像
这样的东西Map(1 <-> "one", 2 <-> "two", 3 <-> "three")
Scala有这样的设施吗?
实际上我还需要更多:3列在三元组中选择三元组中的任何一条(在整个地图中永远不会遇到单个值)。但是2列双向映射也可以提供帮助。
答案 0 :(得分:10)
我的BiMap方法:
object BiMap {
private[BiMap] trait MethodDistinctor
implicit object MethodDistinctor extends MethodDistinctor
}
case class BiMap[X, Y](map: Map[X, Y]) {
def this(tuples: (X,Y)*) = this(tuples.toMap)
private val reverseMap = map map (_.swap)
require(map.size == reverseMap.size, "no 1 to 1 relation")
def apply(x: X): Y = map(x)
def apply(y: Y)(implicit d: BiMap.MethodDistinctor): X = reverseMap(y)
val domain = map.keys
val codomain = reverseMap.keys
}
val biMap = new BiMap(1 -> "A", 2 -> "B")
println(biMap(1)) // A
println(biMap("B")) // 2
当然,可以为<->
而不是->
添加语法。
答案 1 :(得分:10)
Guava有一个bimap可以和
一起使用import scala.collection.JavaConversions._
答案 2 :(得分:3)
这是Guava BiMap
的快速Scala包装器。
import com.google.common.{collect => guava}
import scala.collection.JavaConversions._
import scala.collection.mutable
import scala.languageFeature.implicitConversions
class MutableBiMap[A, B] private (
private val g: guava.BiMap[A, B] = new guava.HashBiMap[A, B]()) {
def inverse: MutableBiMap[B, A] = new MutableBiMap[B, A](g.inverse)
}
object MutableBiMap {
def empty[A, B]: MutableBiMap[A, B] = new MutableBiMap()
implicit def toMap[A, B] (x: MutableBiMap[A, B]): mutable.Map[A,B] = x.g
}
答案 3 :(得分:2)
我在Scala中有一个非常简单的BiMap
:
case class BiMap[A, B](elems: (A, B)*) {
def groupBy[X, Y](pairs: Seq[(X, Y)]) = pairs groupBy {_._1} mapValues {_ map {_._2} toSet}
val (left, right) = (groupBy(elems), groupBy(elems map {_.swap}))
def apply(key: A) = left(key)
def apply[C: ClassTag](key: B) = right(key)
}
用法:
val biMap = BiMap(1 -> "x", 2 -> "y", 3 -> "x", 1 -> "y")
assert(biMap(1) == Set("x", "y"))
assert(biMap("x") == Set(1, 3))
答案 4 :(得分:1)
我不认为它是开箱即用的,因为通用行为不易提取
如何处理与clean api中的几个键匹配的值?
但是对于特定情况,这是一个很好的练习,可能有所帮助。必须更新它,因为没有使用散列并且获取键或值是O(n)。
但是这个想法是让你写一些类似于你建议的东西,但使用Seq而不是Map ......
在隐式和特征的帮助下,加上查找,您可以使用一种干净的api(fromKey
,fromValue
)来模拟您需要的内容。
具体是一个值不应该出现在几个地方......至少在这个实现中。
trait BiMapEntry[K, V] {
def key:K
def value:V
}
trait Sem[K] {
def k:K
def <->[V](v:V):BiMapEntry[K, V] = new BiMapEntry[K, V]() { val key = k; val value = v}
}
trait BiMap[K, V] {
def fromKey(k:K):Option[V]
def fromValue(v:V):Option[K]
}
object BiMap {
implicit def fromInt(i:Int):Sem[Int] = new Sem[Int] {
def k = i
}
implicit def fromSeq[K, V](s:Seq[BiMapEntry[K, V]]) = new BiMap[K, V] {
def fromKey(k:K):Option[V] = s.find(_.key == k).map(_.value)
def fromValue(v:V):Option[K] = s.find(_.value == v).map(_.key)
}
}
object test extends App {
import BiMap._
val a = 1 <-> "a"
val s = Seq(1 <-> "a", 2 <-> "b")
println(s.fromKey(2))
println(s.fromValue("a"))
}
答案 5 :(得分:0)
Scala是不可变的,值被指定为引用而不是复制,因此内存占用仅用于引用/指针存储,最好用于两个映射,其中类型A为第一个键,而类型为B为第二个键分别映射到B和A,而不是映射的时间交换。并且交换实现也有它自己的内存占用,并且新交换的哈希映射也将存在于内存中,直到执行父回调和垃圾收集器调用。如果经常需要交换地图,那么实际上你使用的内存与初始的两个地图实现相同或更多的内存。
您可以尝试使用单一地图的另一种方法是这种方法(仅适用于使用映射值获取密钥):
def getKeyByValue[A,B](map: Map[A,B], value: B):Option[A] = hashMap.find((a:A,b:B) => b == value)
按密钥查找Scala的代码:
/** Find entry with given key in table, null if not found.
*/
@deprecatedOverriding("No sensible way to override findEntry as private findEntry0 is used in multiple places internally.", "2.11.0")
protected def findEntry(key: A): Entry =
findEntry0(key, index(elemHashCode(key)))
private[this] def findEntry0(key: A, h: Int): Entry = {
var e = table(h).asInstanceOf[Entry]
while (e != null && !elemEquals(e.key, key)) e = e.next
e
}