在clojure中实现双向地图的最佳方法是什么? (通过双向映射,我的意思是一个关联映射,它可以提供A-> B和B-> A访问。所以实际上,值本身将是向相反方向前进的键。)
我想我可以设置两个地图,每个方向一个,但有没有更惯用的方法呢?
我对我们想要一个双射的情况感兴趣,暗示没有两个键可以映射到相同的值,以及没有强加该条件的情况。
答案 0 :(得分:13)
您可以随时使用Java库,例如Apache commons中的一个集合。 TreeBidiMap实现了java.util.Map
,因此它可以毫不费力地进行选择。
user> (def x (org.apache.commons.collections.bidimap.TreeBidiMap.))
#'user/x
user> (.put x :foo :bar)
nil
user> (keys x)
(:foo)
user> (.getKey x :bar)
:foo
user> (:foo x)
:bar
user> (map (fn [[k v]] (str k ", " v)) x)
(":foo, :bar")
有些东西不会起作用,比如assoc
和dissoc
,因为它们期望持久集合和TreeBidiMap是可变的。
如果你真的想在本地Clojure中这样做,你可以使用元数据来保存反向哈希。这仍然会使内存需求增加一倍,并且每次添加和删除的时间都会增加一倍,但查找速度足够快,至少所有内容都捆绑在一起。
(defn make-bidi []
(with-meta {} {}))
(defn assoc-bidi [h k v]
(vary-meta (assoc h k v)
assoc v k))
(defn dissoc-bidi [h k]
(let [v (h k)]
(vary-meta (dissoc h k)
dissoc v)))
(defn getkey [h v]
((meta h) v))
当然,您可能必须实现一堆其他功能才能获得完整功能。不确定这种方法的可行性。
user> (def x (assoc-bidi (make-bidi) :foo :bar))
#'user/x
user> (:foo x)
:bar
user> (getkey x :bar)
:foo
答案 1 :(得分:3)
对于大多数情况,我发现以下工作正常:
(defn- bimap
[a-map]
(merge a-map (clojure.set/map-invert a-map)))
这将简单地将原始地图与倒置地图合并为一个新地图,然后您可以对结果使用常规地图操作。
它快速而且肮脏,但显然如果任何键可能与任何值相同或者a-map
中的值不唯一,则应避免使用此实现。
答案 2 :(得分:1)
我们可以按如下方式重新解决这个问题:
给定一个元组列表([a1 b1] [a2 b2] ...)
,我们希望能够快速查找a
和b
的元组。所以我们想要映射a -> [a b]
和b -> [a b]
。这意味着至少元组[a b]
可以共享。如果我们考虑实现它,那么很快就会发现这是一个内存数据库表,其中列A和B由两列索引。
所以你可以使用一个轻量级的内存数据库。
抱歉,我不熟悉Java平台,足以推荐具体内容。我想到了原因Datomic,但对于这个特定的用例来说,这可能是一种过度杀伤力。另一方面,根据你对Brian的回答的评论,它具有不可变性,对你来说似乎是可取的。