考虑我们有一个包含少量元素的数组A。我们还有另外两个值X和k。我们的任务是从数组A中找到k个最接近X的元素。如果元素X存在于数组中,则它将不会显示在输出中。如果A = [12、16、22、30、35、39、42、45、48、50、53、55、56]且X = 35,则k =4。输出将为30、39、42、45 。
为了解决这个问题,我们将使用二进制搜索方法。使用此我们将获得交叉点。如果找到交叉点的索引,则可以在O(k)时间中打印k个最接近的元素。
#include<iostream>
using namespace std;
int getCrossoverPoint(int arr[], int left, int right, int x) {
if (arr[right] <= x)
return right;
if (arr[left] > x)
return left;
int mid = (left + right)/2;
if(arr[mid] <= x && arr[mid+1] > x)
return mid;
if(arr[mid] < x)
return getCrossoverPoint(arr, mid+1, right, x);
return getCrossoverPoint(arr, left, mid - 1, x);
}
void findKClosestNumbers(int arr[], int x, int k, int n) {
int l = getCrossoverPoint(arr, 0, n-1, x);
int r = l+1;
int count = 0;
if (arr[l] == x) l--;
while (l >= 0 && r < n && count < k) {
if (x - arr[l] < arr[r] - x)
cout << arr[l--] << " ";
else
cout << arr[r++] << " ";
count++;
}
while (count < k && l >= 0){
cout << arr[l--] << " ";
count++;
}
while (count < k && r < n){
cout << arr[r++] << " ";
count++;
}
}
int main() {
int arr[] ={12, 16, 22, 30, 35, 39, 42, 45, 48, 50, 53, 55, 56};
int n = sizeof(arr)/sizeof(arr[0]);
int x = 35, k = 5;
findKClosestNumbers(arr, x, k, n);
}输出结果
39 30 42 45 48