师傅说:
这个的实现和nth_element函数一样
用快排的思想去选取第k大
它排序后,在a[k]前面都是比它小的,后面都是比它大的
题目
标题:快速排序。
以下代码可以从数组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;
}
注意:只填写划线部分缺少的代码,不要抄写已经存在的代码或符号。
题意
这里用的是快排的思想,所以代码看起来很像快排的代码。不过进行了优化,题目要我们填其中的一个语句。
如果这个语句填的不够优,会导致时间复杂度比O(n)高,不满足题目的要求,从而导致错误。
理解
在普通的快速排序中,每次递归都能走到左指针和右指针相遇,并且对于相遇的这个地方,在它左边的都比它小,在它右边的都比它大。左边和右边不一定有序,然后一直递归下去,直到快速排序执行(递归)到最后一次,才是排序好的。
而这里不一样,这里执行结束时,只能保证我们要求的第K大的值在数组的第K位,在这个位置的左边的都比它小,右边的都比它大,但整个数组不一定是有序的。
int p = rand() % (r - l + 1) + l;
int x = a[p];
{int t = a[p]; a[p] = a[r]; a[r] = t;}//这一步很关键
上面的这段代码把随机取到的那个数和最右边的数进行交换,为后面while循环里面的交换做了铺垫。
这就是为什么后面的交换是a[j]=a[i];以及a[i]=a[j];(我就是这里没看懂)我以为这样操作会把数组中的一些数字覆盖掉,其实并不会的。(可以试着手动模拟一下)
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++;
}
}
这个O(n)算法的实现跟STL里面的nth-element函数是一样的。
https://www.cnblogs.com/yxwkf/p/5233716.html
答案
最后要填的是 a, p+1, r, k-(i-l+1)
感悟
1、算法真奇妙,短短两三句话的修改能让所需时间大为缩短。
2、要多问师傅,不要觉得会打扰到师傅(不要太频繁、问太低级的就好)这里自己一人看了大半天没看懂,师傅三两句话就让我get到点了。
参考了
https://www.cnblogs.com/fisherss/p/10173400.html
这里面的代码写的注释非常详细。