确定S中最接近中位数的k个元素
(算法导论第三版9.3-7题)
时间复杂度O(n)
vector<int> k_elements_closest_to_median(int *array,int start,int end,int k)
{
vector<int> result;
int median = select(array,start,end,(end - start + 1)/2 + (end - start + 1) % 2);
int *array_temp = new int [end - start + 1];
int absolute_value;
for (int i = start; i < end + 1; ++i) {
absolute_value = array[i] - median;
;
array_temp[i-start] = absolute_value >= 0 ? absolute_value: -absolute_value;
}
int k_smallest = select(array_temp,0,end - start,k);
for (int i = start; i < end + 1; ++i) {
absolute_value = array[i] - median;
absolute_value = absolute_value >= 0 ? absolute_value: -absolute_value;
if(absolute_value <= k_smallest)
result.push_back(array[i]);
}
delete[] array_temp;
return result;
}
辅助函数select
链接地址