题目描述
以下代码可以从数组a[]中找出第k小的元素。
它使用了类似快速排序中的分治算法,期望时间复杂度是O(N)的。
请仔细阅读分析源码,填写划线部分缺失的内容。
注意:只提交划线部分缺少的代码,不要抄写任何已经存在的代码或符号。
import java.util.Random;
public class Main{
public static int quickSelect(int a[], int l, int r, int k) {
Random rand = new Random();
int p = rand.nextInt(r - l + 1) + l;
int x = a[p];
int tmp = a[p]; a[p] = a[r]; a[r] = tmp;
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 quickSelect( _________________________________ ); //填空
else return quickSelect(a, l, i - 1, k);
}
public static void main(String args[]) {
int [] a = {1, 4, 2, 8, 5, 7};
System.out.println(quickSelect(a, 0, 5, 4));
}
}
这道题就是考快排,对于快排不详细介绍,只说一说思想
快速排序的原理是每次选出一个数,然后双指针分别从左右两边遍历,实现把大于这个数的数放在这个数的右边,小于这个数的数放在这个数的左边,即也就是把选出来的这个数放在正确位置上。
代码的写法也有一些巧妙之处:双指针,本题提供的代码,是一开始记录下来选出来的这个数,然后左边开始遍历,直到找到大于这个数的数,(然后左右指针所指的数互换位置,这个是另一种写法)然后直接把这个数赋值给右边指针所指的位置,那么原来的值不是被覆盖了?没关系,因为我已经记录下来原来的值是啥了。同样的,然后右边开始遍历,直到找到小于这个数的数,然后直接把这个数赋值给左边指针所指的位置,那么原来的值不是被覆盖了?没关系,因为此时左指针所指的数本来就是没有用的数,在“然后直接把这个数赋值给右边指针所指的位置”操作后,左指针所指的数已经给右边了,现在这个位置上的数不管用了,所以可以直接覆盖,接下来就是同样的循环了。
本题要找第k小的数,所以在原来快排的基础上加上了判断,当前已经排好的数所处的位置,而“坑”就设置在这里了。
一开始大意写成了
return quickSelect(a, i + 1, r, k); // 填空
后来发现,他真的是咸得,比较的时候
if (i - l + 1 == k)
其实直接写i + 1 == k就可以了,因为l随时在变呀!!!!!
那么原来是i + 1 == k
现在是i - (i + 1) + 1 == k1(因为下一个l换成了i + 1)
那么k1 = k - i - 1;
但是还是不对,这讨厌的题!!!!!!!!!
我们在这一步改变了k,下一次应该是原来的k来进行 k - i - 1运算,但实际上用的是我们改变了的k运算的,所以我们应该加上l,把k还原。
再来理解:现在的k不是一开始的k来,而是我用一开始的k减去当前的l(即在进入这个循环前 k - i - 1,i + 1就是当前循环的l)得到的,所以我要还原原来的k,要加上l,就是这样子,我天!!!!!!!!
再来最后一次
k - i - 1 + l,- i - 1是减去要进入的循环的l,而+l是加上当前循环的l来还原k。
可以自己画个坐标轴来想想,
愿你能理解!!!!!!!
import java.util.Random;
public class okt5快速排序 {
public static int quickSelect(int a[], int l, int r, int k) {
Random rand = new Random();
int p = rand.nextInt(r - l + 1) + l;
int x = a[p];
int tmp = a[p];
a[p] = a[r];
a[r] = tmp;
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)// !!!!!!!!这里其实原本简单一点应该是i + 1 == k
return a[i];
if (i - l + 1 < k)
// return quickSelect(a, i + 1, r, k); // 填空
return quickSelect(a, i + 1, r, k - i - 1 + l); // 填空
else
return quickSelect(a, l, i - 1, k);
}
public static void main(String args[]) {
int[] a = { 1, 4, 2, 8, 5, 7 };
System.out.println(quickSelect(a, 0, 5, 4));
}
}