golang 使用泛型进行分组排序
使用golang 泛型编写出一个通用的分组排序
package main
type GetSorterNum interface {
GetSorterNum() int
}
func SortInt(list []int) []int {
if len(list) > 0 {
sort.SliceStable(list, func(i, j int) bool {
return list[j] > list[i]
})
}
return list
}
func getGroupSorted[T GetSorterNum](data []T) [][]T {
// 按照 sort_num 进行分组
group := make(map[int][]T)
for _, d := range data {
sortNum := d.GetSorterNum()
if v, ok := group[sortNum]; ok {
group[sortNum] = append(v, d)
} else {
group[sortNum] = []T{d}
}
}
sortNums := make([]int, 0)
for k := range group {
sortNums = append(sortNums, k)
}
sorted := make([][]T, len(group))
for i, item := range SortInt(sortNums) {
sorted[i] = group[item]
}
return sorted
}
type Kk struct {
Sort int
}
func (k Kk) GetSorterNum() int {
return k.Sort
}
func main() {
a := []Kk{
{
Sort: 4,
},
{
Sort: 4,
},
{
Sort: 99,
},
{
Sort: 99,
},
{
Sort: 11,
},
{
Sort: 11,
},
{
Sort: 1,
},
{
Sort: 1,
},
{
Sort: 992,
},
{
Sort: 18,
},
}
c := getGroupSorted(a)
fmt.Println(c)
}