658. 找到 K 个最接近的元素

658. 找到 K 个最接近的元素

给定一个 排序好 的数组 arr ，两个整数 k 和 x ，从数组中找到最靠近 x（两数之差最小）的 k 个数。返回的结果必须要是按升序排好的。

整数 a 比整数 b 更接近 x 需要满足：

• |a - x| < |b - x| 或者
• |a - x| == |b - x| 且 a < b

示例 1：

输入：arr = [1,2,3,4,5], k = 4, x = 3
输出：[1,2,3,4]

示例 2：

输入：arr = [1,2,3,4,5], k = 4, x = -1
输出：[1,2,3,4]

提示：

• 1 <= k <= arr.length
• 1 <= arr.length <= 104
• arr 按 升序 排列
• -104 <= arr[i], x <= 104

func findClosestElements(arr []int, k, x int) []int {
    // 稳定排序，在绝对值相同的情况下，保证更小的数排在前面
    sort.SliceStable(arr, func(i, j int) bool { return abs(arr[i]-x) < abs(arr[j]-x) })
    arr = arr[:k]
    sort.Ints(arr)
    return arr
}

func abs(x int) int {
    if x < 0 {
        return -x
    }
    return x
}

golang中的sort包的使用

关键点：

func SliceStable(slice interface{}, less func(i int, j int) bool)有两个参数：

第一个：slice interface{}是切片的接口

第二个：less func(i int, j int) bool是一个less函数，less函数需要自行定义