go sorted map排序映射|极客笔记

go sorted map排序映射

在Go语言中，map是一种无序的集合类型，它由一组键-值对组成。在操作map时，我们通常是按照键的插入顺序来遍历。但有时候我们希望按照键的顺序进行遍历或排序，这时就需要使用排序映射。

什么是排序映射

排序映射是一种将map按照键或值进行排序的数据结构。在Go语言中，我们可以使用自定义结构体或第三方库来实现排序映射。这里我们介绍一种简单的方法来实现排序映射，即使用slice对map进行排序。

使用slice对map进行排序

下面是一个简单的示例代码，演示了如何使用slice对map进行排序：

package main

import (
    "fmt"
    "sort"
)

func main() {
    m := map[string]int{
        "apple":  10,
        "banana": 5,
        "cherry": 20,
        "date":   8,
    }

    keys := make([]string, 0, len(m))
    for key := range m {
        keys = append(keys, key)
    }

    sort.Strings(keys)

    for _, key := range keys {
        fmt.Println(key, m[key])
    }
}

在上面的示例中，我们首先创建了一个包含map键的slice keys，然后使用sort.Strings()函数对slice进行排序。最后，我们按照排序后的键顺序遍历map，并输出键值对。

运行上面的代码，我们可以得到按键排序后的输出：

apple 10
banana 5
cherry 20
date 8

使用自定义结构体排序map

除了使用slice对map进行排序外，我们还可以创建一个自定义结构体来实现对map的排序。下面是一个示例代码：

package main

import (
    "fmt"
    "sort"
)

type Pair struct {
    Key   string
    Value int
}

type PairList []Pair

func (p PairList) Len() int           { return len(p) }
func (p PairList) Less(i, j int) bool { return p[i].Value < p[j].Value }
func (p PairList) Swap(i, j int)      { p[i], p[j] = p[j], p[i] }

func main() {
    m := map[string]int{
        "apple":  10,
        "banana": 5,
        "cherry": 20,
        "date":   8,
    }

    pairs := make(PairList, 0, len(m))
    for key, value := range m {
        pairs = append(pairs, Pair{key, value})
    }

    sort.Sort(pairs)

    for _, pair := range pairs {
        fmt.Println(pair.Key, pair.Value)
    }
}

在上面的示例中，我们创建了一个Pair结构体来表示键值对，然后创建了一个PairList类型来存储多个Pair结构体。我们还定义了PairList类型的Len()、Less()和Swap()方法来实现对结构体的排序。

通过运行上面的代码，我们可以得到按值排序后的输出：

banana 5
date 8
apple 10
cherry 20

使用第三方库实现排序映射

除了自定义结构体和使用slice，我们还可以使用第三方库来实现排序映射。其中比较常用的库有github.com/bradfitz/slice和github.com/emirpasic/gods/maps/treemap。

下面是一个使用github.com/bradfitz/slice库来实现对map按值排序的示例代码：

package main

import (
    "fmt"
    "sort"

    "github.com/bradfitz/slice"
)

func main() {
    m := map[string]int{
        "apple":  10,
        "banana": 5,
        "cherry": 20,
        "date":   8,
    }

    keys := make([]string, 0, len(m))
    values := make([]int, 0, len(m))

    for key, value := range m {
        keys = append(keys, key)
        values = append(values, value)
    }

    slice.Sort(values, func(i, j int) bool {
        return values[i] < values[j]
    })

    for _, value := range values {
        for key, v := range m {
            if v == value {
                fmt.Println(key, value)
                break
            }
        }
    }
}

通过引入github.com/bradfitz/slice库，我们可以使用slice.Sort()函数来对值进行排序。然后我们可以按照排序后的值找到对应的键，并输出键值对。