GOlang 对象排序

本贴最后更新于 2212 天前,其中的信息可能已经水流花落

golang sort 包

包中这样介绍.

type Interface

type Interface interface {
	// Len方法返回集合中的元素个数
	Len() int
	// Less方法报告索引i的元素是否比索引j的元素小
	Less(i, j int) bool
	// Swap方法交换索引i和j的两个元素
	Swap(i, j int)
}

一个满足 sort.Interface 接口的(集合)类型可以被本包的函数进行排序。方法要求集合中的元素可以被整数索引。

golang 中排序只需要重写 len, less, swap 方法

func Sort

func Sort(data Interface)

它调用 1 次 data.Len 确定长度,调用 O(n*log(n))次 data.Less 和 data.Swap。本函数不能保证排序的稳定性(即不保证相等元素的相对次序不变)。

使用

结构体

type SortJson struct {
	Name  string `json:"name"`
	Count int    `json:"count"`
}

重写 Len(), Swap 方法

type SortJsonSlice []SortJson

func (s SortJsonSlice) Len() int {
	return len(s)
}

func (s SortJsonSlice) Swap(i, j int) {
	s[i], s[j] = s[j], s[i]
}

func (s SortJsonSlice) Less(i, j int) bool {
	return s[i].Count > s[j].Count
}

test

func TestSortObject(t *testing.T) {
	dj := SortJsonSlice{SortJson{Name: "a", Count: 1}, {Name: "c", Count: 3}, {Name: "b", Count: 2}, {Name: "f", Count: 6}, {Name: "e", Count: 4}, {Name: "g", Count: 7}}
	fmt.Println(dj)
	sort.Sort(dj)
	fmt.Println(dj)
}

这个打印的结果就是

=== RUN   TestSortObject
[{a 1} {c 3} {b 2} {f 6} {e 4} {g 7}]
[{g 7} {f 6} {e 4} {c 3} {b 2} {a 1}]
--- PASS: TestSortObject (0.00s)

排序成功

升级使用

上面的写就已经很方便了, 但是有个问题. 如果你可能需要多种排序方式(正序, 倒序, 时间, 数量). 就会很烦.

可以在次重写上面重写过的数据 只重写 Less 方法.

比如, 根据 count 正序. (上面的方法是倒序)

// 将 Less 绑定到 OrderCount 上
type OrderCount struct {
	SortJsonSlice
}

// 将 DeviceJsonSlice 包装起来到 ByWeight 中
func (s OrderCount) Less(i, j int) bool {
	return s.SortJsonSlice[i].Count < s.SortJsonSlice[j].Count
}

如果要添加一个排序方式 根据 Name 排序:

type OrderName struct {
	SortJsonSlice
}

// 将 DeviceJsonSlice 包装起来到 ByWeight 中
func (s OrderCount) Less(i, j int) bool {
	return s.SortJsonSlice[i].Name < s.SortJsonSlice[j].Name
}

这样只要给每种排序方式重写一个 Less 方法就好了.

test

func TestSortObject(t *testing.T) {
	dj := OrderCount{[]DeviceJson{{Name: "a", Count: 1}, {Name: "c", Count: 3}, {Name: "b", Count: 2}, {Name: "f", Count: 6}, {Name: "e", Count: 4}, {Name: "g", Count: 7}}}
	fmt.Println(dj)
	sort.Sort(dj)
	fmt.Println(dj)
}
=== RUN   TestSortObject
{[{a 1} {c 3} {b 2} {f 6} {e 4} {g 7}]}
{[{a 1} {b 2} {c 3} {e 4} {f 6} {g 7}]}
--- PASS: TestSortObject (0.00s)
PASS
  • golang

    Go 语言是 Google 推出的一种全新的编程语言,可以在不损失应用程序性能的情况下降低代码的复杂性。谷歌首席软件工程师罗布派克(Rob Pike)说:我们之所以开发 Go,是因为过去 10 多年间软件开发的难度令人沮丧。Go 是谷歌 2009 发布的第二款编程语言。

    497 引用 • 1387 回帖 • 294 关注

相关帖子

欢迎来到这里!

我们正在构建一个小众社区,大家在这里相互信任,以平等 • 自由 • 奔放的价值观进行分享交流。最终,希望大家能够找到与自己志同道合的伙伴,共同成长。

注册 关于
请输入回帖内容 ...