'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，然后根据绝对值排序，绝对值小的在前面

代码

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22

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; } };

-------------end of file thanks for reading-------------