我正在尝试实现quickselect算法 . 虽然,我已经理解了它背后的理论;我发现很难把它转换成一个运作良好的程序 .
以下是我如何逐步实施它以及我面临的问题:
Problem: 找到A [] = {2,1,3,7,5,4,6}中的第4个最小元素
k = 4 .
指数: 0|1|2|3|4|5|6
对应值: 2|1|3|7|5|4|6
最初, l = 0 和 r = 6
Step 1) 将pivot作为最左边的元素(pivot将始终是此问题中最左侧的) -
pivot_index = 0
pivot_value = 2
Step 2) 应用分区算法;将枢轴放在正确的位置( [
p] ) -
我们得到以下数组: 1|2|3|7|5|4|6
哪里, pivot_index = i-1 = 1
因此, pivot_value = 2
Step 3) 将 pivot_index 与 k 进行比较 -
k=3 , pivot_index = 1 ; k > pivot_index
因此,我们的第k个最小数字位于数组的右侧 .
右数组= i to r ,我们不再打扰左侧部分( l to i-1 )了 .
Step 4) 我们将 k 的值修改为 k - (pivot_index) => 4-1 = 2; k = 3 .
Here is the problem: k 的值不应该是2吗?因为我们在数组的左侧部分有两个值: 1|2 ?我们应该将 k 计算为 k - (pivot_index+1) 吗?
我们假设 k = 3 是正确的 .
Step 5) "New"要处理的数组: 3|7|5|4|6 ,带有相应的索引: 2|3|4|5|6
现在, pivot_index = 2 和 pivot_index = 3
Step 6) 在上述阵列上应用分区算法 -
3|7|5|4|6 (数组保持不变,因为数据库本身是最低值) . i = 3
pivot_index = i-1 = 2 pivot_value = 3
Step 7) 将 pivot_index 与 k 进行比较
k=3 和 pivot_index=2
k > pivot_index
等等....
Is this approach correct?
这是我的代码 not 正在工作 . 我使用随机数生成器来选择随机数,然后使用数组中的第一个元素交换数据透视表 .
#include
#include
void print_array(int arr[], int array_length){
int i;
for(i=0; i
printf("%d ", arr[i]);
}
}
int random_no(min, max){
int diff = max-min;
return (int) (((double)(diff+1)/RAND_MAX) * rand() + min);
}
void swap(int *a, int *b){
int temp;
temp = *a;
*a = *b;
*b = temp;
}
int get_kth_small(int arr[], int k, int l, int r){
if((r-l) >= 1){
k = k + (l-1);
int pivot_index = random_no(l, r);
int i, j;
swap(&arr[pivot_index], &arr[l]); //Switch the pivot with the first element in the array. Now, the pivit is in arr[l]
i=l+1;
for(j=l+1; j<=r; ++j){
if(arr[j]
swap(&arr[j], &arr[i]);
++i;
}
}
swap(&arr[l], &arr[i-1]); //Switch the pivot to the correct place;
p
printf("value of i-1: %d\n", i-1);
printf("Value of k: %d\n", k);
if(k == (i-1)){
printf("Found: %d\n", arr[i]);
return 0;
}
if(k>(i-1)){
k=k-(i-1);
get_kth_small(arr, k, i, r);
} else {
get_kth_small(arr, k, l, r-1);
}
//get_kth_small(arr, k, i, r);
//get_kth_small(arr, k, l, i-1);
}
}
void main(){
srand(time(NULL));
int arr[] = {2,1,3,7,5,4,6};
int arr_size = sizeof(arr)/sizeof(arr[0]);
int k = 3, l = 0;
int r = arr_size - 1;
//printf("Enter the value of k: ");
//scanf("%d", &k);
get_kth_small(arr, k, l, r);
print_array(arr, arr_size);
printf("\n");
}