lintcode 在排序数组中找最接近的K个数

描述

给一个目标数 target, 一个非负整数 k, 一个按照升序排列的数组 A。在A中找与target最接近的k个整数。返回这k个数并按照与target的接近程度从小到大排序，如果接近程度相当，那么小的数排在前面。

k是一个非负整数，并且总是小于已排序数组的长度。
给定数组的长度是正整数, 不会超过 10^4
数组中元素的绝对值不会超过 10^4

样例

样例 1:

输入: A = [1, 2, 3], target = 2, k = 3
输出: [2, 1, 3]
样例 2:

输入: A = [1, 4, 6, 8], target = 3, k = 3
输出: [4, 1, 6]

思路

先把每个数减少target，然后根据绝对值排序，绝对值小的在前面

代码

class Solution {
public:
    /*
     * @param A: an integer array
     * @param target: An integer
     * @param k: An integer
     * @return: an integer array
     */
    bool static cmp(int A, int B) {
        return abs(A) < abs(B) ? true : abs(A) == abs(B) ?  A < B : false;
    }
    vector<int> kClosestNumbers(vector<int> &A, int target, int k) {
        // write your code here
        for (int i = 0; i < A.size(); i++)
            A[i] -= target;
        sort(A.begin(), A.end(), cmp);
        vector<int> res(A.begin(), A.begin()+k);
        for (int i = 0; i < res.size(); i++)
            res[i] += target;
        return res;
    }
};