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)
}

posted on 2024-06-14 15:20  biwentao  阅读(17)  评论(0编辑  收藏  举报

导航