今天刷力扣215.数组中第K个最大元素,题目要求用时间复杂度O(n)。因此跟着左程云算法课23复习了随机快速排序。
随机快速排序的核心思想就是在数组中随机选一个值x,把<=x的值移到左边,>x的值移动到右边。x这个值必须在左边区间的最左边,也就是确定了x的位置就是这里,然后再分别递归调用对左右两边进行排序。
一、经典随机快速排序
经典的随机快速排序,在QuickSort中先选定随机数x,然后partition返回x所在的位置下标。在partition过程中,设置一个a表示左边界,xi表示x的位置,最后要和a-1进行交换。下标<a的区间是<=x的区间,最后返回a-1。
java
public void QuickSort1(int[] nums,int l,int r){
if(l>=r){
return;
}
int x=nums[l+ran.nextInt(r-l+1)];
int mid=partition1(nums,l,r,x);
QuickSort1(nums,l,mid-1);
QuickSort1(nums,mid+1,r);
}
public int partition1(int[] nums,int l,int r,int x){
int a=l;
int xi=-1;
for(int i=l;i<=r;i++){
if(nums[i]<=x){
swap(nums,a,i);
if(nums[a]==x){
xi=a;
}
a++;
}
}
if(xi!=-1){
swap(nums,a-1,xi);
}
return a-1;
}
二、荷兰问题优化的随机快速排序
在经典的随机快速排序中,有个不好的点就是如果一个数组中右多个相同的值x,那一次排序后只能确定一个x的位置,其他x还要继续参与排序。荷兰问题优化的思路在于,在partition的时候,将整个数组分成三部分,左边<x,中间=x,右边>x。那就需要设置两个边界值a和b,a表示左区间的越界值,b表示右区间的越界值。
java
public static Random ran=new Random();
public static int left;
public static int right;
public void QuickSort2(int[] nums,int l,int r){
if(l>=r){
return;
}
int x=nums[l+ran.nextInt(r-l+1)];
partition2(nums,l,r,x);
int left1=left;
int right1=right;
QuickSort2(nums,l,left1-1);
QuickSort2(nums,right1+1,r);
}
public void partition2(int[] nums,int l,int r,int x){
int a=l;
int b=r;
int i=l;
while(i<=b){
if(nums[i]<x){
swap(nums,a,i);
a++;
i++;
}
else if(nums[i]==x){
i++;
}
else{
swap(nums,b,i);
b--;
}
}
left=a;
right=b;
}