go排序后索引

最近碰上了一个需求,要得到排序后的原索引序列。

我又不希望自己重新实现一快排出来,所以在接口上重新封装了一下。

package main

import (
"fmt"
"sort"
) type SortIndexs struct {
sort.IntSlice // 可以替换成其他实现了 sort.Interface
indexs []int
} func (p *SortIndexs) Swap(i, j int) {
p.IntSlice.Swap(i, j)
p.indexs[i], p.indexs[j] = p.indexs[j], p.indexs[i]
} func NewSortIndexs(arr []int) *SortIndexs {
s := &SortIndexs{IntSlice: sort.IntSlice(arr), indexs: make([]int, len(arr))}
for i := range s.indexs {
s.indexs[i] = i // 原有排序 indexs
}
return s
} func main() {
s := NewSortIndexs([]int{5,8,10,2,9,6})
sort.Sort(s)
fmt.Println(s.indexs)
fmt.Println(s.IntSlice)
}

输出

[3 0 5 1 4 2]
[2 5 6 8 9 10]
上一篇:乐在其中设计模式(C#) - 策略模式(Strategy Pattern)


下一篇:设计模式之策略模式和状态模式(strategy pattern & state pattern)