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]

转载于:https://www.cnblogs.com/hutusheng/p/6849025.html