标题:快速排序。
以下代码可以从数组a[]中找出第k小的元素。
它使用了类似快速排序中的分治算法,期望时间复杂度是O(N)的。
请仔细阅读分析源码,填写划线部分缺失的内容。
#include <stdio.h>
int quick_select(int a[], int l, int r, int k) {
int p = rand() % (r - l + 1) + l;
int x = a[p];
{int t = a[p]; a[p] = a[r]; a[r] = t;}
int i = l, j = r;
while(i < j) {
while(i < j && a[i] < x) i++;
if(i < j) {
a[j] = a[i];
j--;
}
while(i < j && a[j] > x) j--;
if(i < j) {
a[i] = a[j];
i++;
}
}
a[i] = x;
p = i;
if(i - l + 1 == k) return a[i];
if(i - l + 1 < k) return quick_select( _____________________________ ); //填空
else return quick_select(a, l, i - 1, k);
}
int main()
{
int a[] = {1, 4, 2, 8, 5, 7, 23, 58, 16, 27, 55, 13, 26, 24, 12};
printf("%d\n", quick_select(a, 0, 14, 5));
return 0;
}
解题思路:题意要找到第k小的数,又用分治的思想,即二分法来查找,只有一个填空,填空上面,直接返回了一个数,即找到了答案,所以i-l+1就是当前查看的位置,填空的条件的i-l+1 < k 说明要在右边查找,那么在右边部分的序列就不是第k小的数了,还要减去左边部分的长度,因为左边全部都小于右边, 即quick_select(a, i+1, r, k-(i-l+1))
#include<stdio.h>
#include<stdlib.h>
int quick_select(int a[], int l, int r, int k){
int p = rand()%(r-l+1)+l;
int x = a[p];
int t = a[p]; a[p] = a[r]; a[r] = t;
int i = l, j = r;
while(i < j){
while(i<j && a[i]<x)i++;
if(i<j){
a[j] = a[i];
j--;
}
while(i<j && a[j]>x)j--;
if(i<j){
a[i] = a[j];
i++;
}
}
a[i] = x;
p = i;
for(int s=0; s<14; ++s){
printf("%d ", a[s]);
}
printf("\n%d %d\n", i, j);
// return 0;
if(i-l+1 == k)return a[i];
if(i-l+1<k)return quick_select(a, i+1, r, k-i+l-1); //填空
else return quick_select(a, l, i-1, k);
}
int main()
{
int a[] = {1, 4, 2, 8, 5, 7, 23, 58, 16, 27, 55, 13, 26, 24, 12};
printf("%d\n", quick_select(a, 0, 14, 5));
return 0;
}