如何在 Swift 中实现双向映射

Question

我目前需要一个高性能的双向地图。在 Swift 中，字典可以反转，但是，它将返回它所组成的类型的元组，而不是对应的字典。

是否有一个库，或者有人对如何解决这个问题有想法？

谢谢

Answer 1

使用 Swift 4，您可以使用泛型结构体轻松创建自己的结构：

struct BidiMap<F:Hashable,T:Hashable>
{
   private var _forward  : [F:T]? = nil
   private var _backward : [T:F]? = nil
   var forward:[F:T]  
   { 
      mutating get 
      { 
        _forward = _forward ?? [F:T](uniqueKeysWithValues:_backward?.map{($1,$0)} ?? [] ) 
        return _forward!
      }
      set { _forward = newValue; _backward = nil }
   }
   var backward:[T:F]  
   { 
      mutating get 
      { 
        _backward = _backward ?? [T:F](uniqueKeysWithValues:_forward?.map{($1,$0)} ?? [] ) 
        return _backward!
      }
      set { _backward = newValue; _forward = nil }
   }
   init(_ dict:[F:T] = [:])
   { forward = dict  }
   init(_ values:[(F,T)])
   { forward = [F:T](uniqueKeysWithValues:values) }
   subscript(_ key:T) -> F? 
   { mutating get { return backward[key] } set{ backward[key] = newValue } }
   subscript(_ key:F) -> T?
   { mutating get { return forward[key]  } set{ forward[key]  = newValue } }
   subscript(to key:T) -> F? 
   { mutating get { return backward[key] } set{ backward[key] = newValue } }
   subscript(from key:F) -> T?
   { mutating get { return forward[key]  } set{ forward[key]  = newValue } }
   var count:Int { return _forward?.count ?? _backward?.count ?? 0 }
}
var bd = BidiMap( [1:"A", 2:"B", 3:"C"] )
bd[1] // "A"
bd["B"] // 2
bd[4] = "D"
bd[to:"D"] // 4
bd[from:4] // "D"
var int2int = BidiMap( [1:2, 5:3] )
int2int[from:1] // 2
int2int[to:3] // 5

[编辑] 通过延迟重建镜像字典直到实际引用它，提高了性能。