Go 自定義排序

sort

Go語言包中包含了排序包 sort，其中針對一般的strings、int類型已經有了排序方法

sort.Ints(a []int)
sort.Strings(a []string)

自定義排序

1、分別實現三個函數

func (p MySlice) Len() int
func (p MySlice) Less(i, j int) bool
func (p MySlice) Swap(i, j int)

然後調用

sort.Sort(data sort.Interface)

示例如下

package main

import (
	"fmt"
	"sort"
)
type MyDefSlice struct {
	Id   int
	Name string
}
type MySlice []MyDefSlice

func (p MySlice) Less(i, j int) bool {
	return p[i].Id > p[j].Id
}
func (p MySlice) Len() int {
	return len(p)
}
func (p MySlice) Swap(i, j int) {
	p[i], p[j] = p[j], p[i]
}

func main() {
	mySlice := make(MySlice, 0)
	for i := 0; i < 10; i++ {
		mySlice = append(mySlice, MyDefSlice{
			Id:   i,
			Name: fmt.Sprintf("slice_%d", i),
		})
	}
	sort.Sort(mySlice)
	fmt.Printf("mySlice: %v\n", mySlice)
}

執行結果

mySlice: [{9 slice_9} {8 slice_8} {7 slice_7} {6 slice_6} {5 slice_5} {4 slice_4} {3 slice_3} {2 slice_2} {1 slice_1} {0 slice_0}]

2、調用sort.Slice並實現less函數

sort.Slice(slice interface{}, less func(i int,j int) bool)

示例如下

package main

import (
	"fmt"
	"sort"
)
type MyDefSlice struct {
	Id   int
	Name string
}
type MySlice []MyDefSlice

func main() {
	mySlice := make(MySlice, 0)
	for i := 0; i < 10; i++ {
		mySlice = append(mySlice, MyDefSlice{
			Id:   i,
			Name: fmt.Sprintf("slice_%d", i),
		})
	}
	sort.Slice(mySlice, func(i, j int) bool { return mySlice[i].Id > mySlice[j].Id })
	fmt.Printf("mySlice: %v\n", mySlice)
}

執行結果

mySlice: [{9 slice_9} {8 slice_8} {7 slice_7} {6 slice_6} {5 slice_5} {4 slice_4} {3 slice_3} {2 slice_2} {1 slice_1} {0 slice_0}]