在golang中排序int切片

时间:2018-07-31 21:03:08

标签: sorting go slice

假设我有一个包含以下数字的整数切片:2781,4706,1347,1192,3087,2920,198,2312,930,3284,1677,3687,2011,4107,4148,4195,2159,1263, 2918,2503

我想根据它们与某个值的距离有多近对它们进行排序。 例如,假设我的值为1200。

经过排序后,我的切片将是:1192,1263,1347,1677,2011,2159,2312,2503,2781,2918,2920,3087,3284,3687,4107,4148,4195,4706,930,198

我尝试了以下方法:遍历数组,减去指定的值,然后以升序排序,最终再次将值相加,以返回到原始数。这种方法并非在所有测试用例中都有效,而且由于对数组进行了变异,因此我知道它的坏习惯。

1 个答案:

答案 0 :(得分:4)

一种方法是将sort.Slice(...)与“较少”(比较器)函数一起使用,该函数简单地返回两个给定值与目标值之间的较小距离。

例如(Go Playground):

func main() {
  xs := []int{2781, 4706, 1347, 1192, 3087, 2920, 198, 2312, 930, 3284, 1677, 3687, 2011, 4107, 4148, 4195, 2159, 1263, 2918, 2503}
  sortByDistanceFrom(1200, xs)
  fmt.Println(xs)
  // [1192 1263 1347 930 1677 2011 2159 198 2312 2503 2781 2918 2920 3087 3284 3687 4107 4148 4195 4706]

}

func sortByDistanceFrom(x int, xs []int) {
  sort.Slice(xs, func(i, j int) bool {
    di := math.Abs(float64(x - xs[i]))
    dj := math.Abs(float64(x - xs[j]))
    return di < dj
  })
}

请注意,此示例返回的第四个值是930,而不是示例输出产生的1677,因为它计算了到目标数(|1200-1677|=433 > |1200-930|=270)的绝对距离。如果您希望大于目标值而不是小于目标值,则必须相应地修改比较器功能。