在Golang中以<value,key>格式反转地图

Question

我的程序地图如下所示:

fruit_map := map[string]string {
    "apple": "likey",
    "orange": "no likey",
}

我想反转它,以便它读取以下内容:

{
    "likey": "apple",
    "no likey": "orange",
}

值中没有重复项.此外,我的地图很小 - 约200键.我没有找到任何内置方法来反转像这样的地图.有没有办法快速做到这一点？我对空间复杂性并不感到困扰,但解决方案需要快速.

谢谢.

Answer 1

您可以编写一个for循环来迭代原始地图的键值对，并将它们放入新地图中（请参阅函数reverseMap）

代码@https://play.golang.org/p/5y1gABTpdb8

package main

import (
    "fmt"
)

func main() {
    fruit_map := map[string]string{
        "apple":  "likey",
        "orange": "no likey",
    }

    reversedMap := reverseMap(fruit_map)
    fmt.Println(reversedMap)
}

func reverseMap(m map[string]string) map[string]string {
    n := make(map[string]string, len(m))
    for k, v := range m {
        n[v] = k
    }
    return n
}

输出：

map[likey:apple no likey:orange]

顺便说一句fruit_map，像fruitMap.

Answer 2

其他答案提供了基于直接处理地图的简单解决方案。

另一种解决方案是将双向映射封装为独立的实用程序，其优点是您可以为其编写彻底的单元测试，然后能够依靠它通过简单的 API 正确运行。

这是我的示例实现（不完整并且还没有必要的单元测试）：

包主

import (
    "fmt"
)

func main() {
    biMap := NewBiMap()
    biMap.Put("apple", "likey")
    biMap.Put("orange", "no likey")
    v, _ := biMap.GetByValue("no likey")
    fmt.Println(v)
}

type BiMap struct {
    ab map[string]string
    ba map[string]string
}

func NewBiMap() *BiMap {
    return &BiMap{make(map[string]string), make(map[string]string)}
}

func (m *BiMap) Put(key, value string) *BiMap {
    m.ab[key] = value
    m.ba[value] = key
    return m
}

func (m *BiMap) GetByKey(key string) (value string, exists bool) {
    value, exists = m.ab[key]
    return
}

func (m *BiMap) GetByValue(value string) (key string, exists bool) {
    key, exists = m.ba[value]
    return
}

func (m *BiMap) Len() int {
    return len(m.ab)
}

func (m *BiMap) DeleteKey(key string) *BiMap {
    value, exists := m.ab[key]
    if exists {
        delete(m.ab, key)
        delete(m.ba, value)
    }
    return m
}

func (m *BiMap) DeleteValue(value string) *BiMap {
    key, exists := m.ba[value]
    if exists {
        delete(m.ab, key)
        delete(m.ba, value)
    }
    return m
}