在这种情况下,最简单的解决方案是像带有垃圾收集器的语言一样工作:
use std::collections::HashMap;
use std::rc::Rc;
use std::hash::Hash;
use std::ops::Deref;
struct BidiMap<A, B> {
left_to_right: HashMap<Rc<A>, Rc<B>>,
right_to_left: HashMap<Rc<B>, Rc<A>>,
}
impl<A, B> BidiMap<A, B>
where
A: Eq + Hash,
B: Eq + Hash,
{
fn new() -> Self {
BidiMap {
left_to_right: HashMap::new(),
right_to_left: HashMap::new(),
}
}
fn put(&mut self, a: A, b: B) {
let a = Rc::new(a);
let b = Rc::new(b);
self.left_to_right.insert(a.clone(), b.clone());
self.right_to_left.insert(b, a);
}
fn get(&self, a: &A) -> Option<&B> {
self.left_to_right.get(a).map(Deref::deref)
}
fn get_reverse(&self, b: &B) -> Option<&A> {
self.right_to_left.get(b).map(Deref::deref)
}
}
fn main() {
let mut map = BidiMap::new();
map.put(1, 2);
println!("{:?}", map.get(&1));
println!("{:?}", map.get_reverse(&2));
}
当然,您会想要更严格的代码,因为这可以让您打破双向映射。这只是向您展示了解决问题的一种方法。
显然,其中之一必须拥有数据
显然,这不是真的^_^。在这种情况下,两个地图使用共享所有权Rc
。
对这个解决方案进行基准测试,以了解它是否足够高效。
做任何更有效的事情都需要对所有权进行更深入的思考。例如,如果left_to_right
地图拥有数据并且您在另一个地图中使用了原始指针,则该指针将在第一个地图重新分配后立即失效。