数据结构
网络
RDBMS
操作系统
Java
MS Excel
iOS
HTML
CSS
Android
Python
C 编程
C++
C#
MongoDB
MySQL
Javascript
PHP
物理
化学
生物
数学
英语
经济学
心理学
社会学
时尚研究
法律研究用 C++ 查找最接近的 K 个元素
假设我们有一个排序数组，还给定两个整数 k 和 x，我们必须在该数组中找到 k 个最接近 x 的元素。结果应按升序排序。如果存在平局，则始终优先考虑较小的元素。因此，如果输入类似于 [1,2,3,4,5]，并且 k = 4，x = 3，则输出将是 [1,2,3,4]
为了解决这个问题，我们将按照以下步骤进行 -
- 创建一个称为 ans 的数组
- 将 low 设置为 0，将 high 设置为数组大小 - k
- 当 low < high
- mid := low + (high - low) /2
- 如果 x - arr[mid] > arr[mid + k] - x，则 low := mid + 1，否则 high := mid
- 对于 low 到 low + k 范围内的每个 i
- 将 arr[i] 插入 ans 数组
- 返回 ans
示例（C++）
让我们看看以下实现以获得更好的理解 -
#include <bits/stdc++.h>
using namespace std;
void print_vector(vector<auto> v){
cout << "[";
for(int i = 0; i<v.size(); i++){
cout << v[i] << ", ";
}
cout << "]"<<endl;
}
class Solution {
public:
vector<int> findClosestElements(vector<int>& arr, int k, int x) {
vector <int> ans;
int low = 0;
int high = arr.size() - k;
while(low < high){
int mid = low + (high - low)/2;
if(x - arr[mid] > arr[mid + k] - x){
low = mid + 1;
}
else high = mid;
}
for(int i = low ; i < low + k ; i++)ans.push_back(arr[i]);
return ans;
}
};
main(){
Solution ob;
vector<int> v = {1,2,3,4,5};
print_vector(ob.findClosestElements(v, 4, 3));
}输入
[1,2,3,4,5] 4 3
输出
[1,2,3,4]
更新于：29-4-2020
371 次浏览
广告