如何按照map的值对slice进行排序

Question

这似乎是一个基本问题，但我找不到简单的答案。

我有一个切片：

[]string{"dog", "cat", "bird"}

如何通过在映射中查找排序值来对其进行排序呢？

map[string]int{"dog": 2, "cat":3, "bird": 1}

因此，切片应按以下顺序排序：

[]string{"bird", "dog", "cat"}

- js837

Answer 1

实现 sort.Interface 接口，用于存储数据和权重的类型：

import "sort"

type WeightedStringSlice struct {
    Strings []string
    Weights map[string]int
}

func (s *WeightedStringSlice) Len() int {
    return len(s.Strings)
}

func (s *WeightedStringSlice) Less(i, j int) bool {
    return s.Weights[s.Strings[i]] < s.Weights[s.Strings[j]]
}

func (s *WeightedStringSlice) Swap(i, j int) {
    s.Strings[i], s.Strings[j] = s.Strings[j], s.Strings[i]
}

接下来在它上面调用 sort.Sort：

data := WeightedStringSlice{
    Strings: []string{"dog", "cat", "bird"},
    Weights: map[string]int{"dog": 2, "cat": 3, "bird": 1},
}
sort.Sort(&data)
fmt.Printf("%v\n", data.Strings)

Live Demo

- user1804599