LintCode 題解丨Google高頻面試題：在排序數組中找最接近的K個數

給一個目标數 ​target​, 一個非負整數 ​k​, 一個按照升序排列的數組 ​A​。在​A​中找與​target​最接近的​k​個整數。傳回這​k​個數并按照與​target​的接近程度從小到大排序，如果接近程度相當，那麼小的數排在前面。

​k​是一個非負整數，并且總是小于已排序數組的長度。

給定數組的長度是正整數, 不會超過 10​^4​​

數組中元素的絕對值不會超過 10​^4

線上評測位址：

LintCode 領扣

樣例 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​, 如果不存在則傳回大于 ​target​ 的最小的或者小于 ​target​ 的最大的元素均可.

然後使用兩根指針從該位置開始向兩端周遊, 每次把內插補點比較小的元素放入答案中然後将該指針向邊界方向移動一下即可.

public class Solution {

/**
 * @param A      an integer array
 * @param target an integer
 * @param k      a non-negative integer
 * @return an integer array
 */
public int[] kClosestNumbers(int[] A, int target, int k) {
    int[] result = new int[k];

    if (A == null || A.length == 0) {
        return A;
    }
    if (k > A.length) {
        return A;
    }

    int index = firstIndex(A, target);

    int start = index - 1;
    int end = index;
    for (int i = 0; i < k; i++) {
        if (start < 0) {
            result[i] = A[end++];
        } else if (end >= A.length) {
            result[i] = A[start--];
        } else {
            if (target - A[start] <= A[end] - target) {
                result[i] = A[start--];
            } else {
                result[i] = A[end++];
            }
        }
    }
    return result;
}

private int firstIndex(int[] A, int target) {
    int start = 0, end = A.length - 1;
    while (start + 1 < end) {
        int mid = start + (end - start) / 2;
        if (A[mid] < target) {
            start = mid;
        } else if (A[mid] > target) {
            end = mid;
        } else {
            end = mid;
        }
    }

    if (A[start] >= target) {
        return start;
    }
    if (A[end] >= target) {
        return end;
    }
    return A.length;
}           

}

更多題解參考：

九章算法 - 幫助更多中國人找到好工作，矽谷頂尖IT企業工程師實時線上授課為你傳授面試技巧

​