Question

似乎是一个基本问题，但无法找到简单的答案。

我有一个片段：

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

通过在地图中查找排序值来排序它的最佳方法是什么：

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

这样切片的排序如下：

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

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

如何按地图的值对切片进行排序

1 个答案: