众所周知,快速排序已经是所有排序中最快的排序了,不然怎么叫快排?但是经典的快速排序还能再优化,实现真正的“快速”二字,当然时间复杂度依然是O(nlogn)
,这是目前排序最快的时间复杂度了。废话不多说,进入正题。
首先要从经典的荷兰国旗问题说起,leetcode上也有相似的问题,链接:Sort Colors,问题描述:给定一个数组,要求将数组分成三个部分,从左到右依次为<0、=0、>0
的数,注:(同一部分不要求有序,如:-2,-3都是小于0,所以可以是-2, -3,也可以是-3,-2。)
例:
输入: [-1,2,0,-2,1,0]
输出:[-1,-2,0,0,1,2]
要求:时间复杂度O(n),空间复杂度O(1)。
一番思考后……
给出解题思路:既然要求分成三个部分,那就定义两个变量left
, right
,分别指向数组的头部和尾部,left
左边的代表<0,和right
右边的代表>0,剩下中间的就是=0部分了。
再定义一个指针p
,从0开始,用于记录遍历到了数组的哪个位置。如果遇到<0的数字,那么当前p
位置的数字与left
位置的数字交换。同时p++,left++
,代表考虑数组下一个数字并且<0的区间扩大一步。而如果遇到>0的数字,那么p
位置的数字与right
位置的数字交换,且--more
, 但p
位置不变。因为从more
位置交换来的数字还未与0做过比较。最后遇到等于0的数字,跳过,即p++
。
思想有了,上代码(Java实现):
class Solution {
public void sortColors(int[] nums) {
int num = 0;
int left = 0, right = nums.length, p = 0;
while(p < right){
if (nums[p] < num){
swap(nums, left++, p++);
} else if(nums[p] > num){
swap(nums, p, --right);
} else {
p++;
}
}
}
public void swap(int[] nums, int i, int j){
int t = nums[i];
nums[i] = nums[j];
nums[j] = t;
}
}
不知你有没有注意,上述的思想跟快排的partition
思想很相似,选一个数num
作为基准,并将原本的数组分为三个部分。即<num、=num、和>num部分。而<num和>num的部分又可以对其进行相同的操作:分成三个小部分,直到数组只剩一个元素,递归结束。下面使用 [-1,2,0,-2,1,0]
来画个图来描述下:
递归实现(Java):
public static void quickSort(int[] a, int L, int R) {
if (L < R - 1) {
// 随机选择L - R 之间的位置的数字作为基准数
int randomPos = (int)(Math.random() * (R - L) + L);
int region[] = partition(a, L, R, a[randomPos]); //与基准数值相等的排好序
quickSort(a, L, region[0]); // 以基准数的左边界排好序
quickSort(a, region[1], R); // 以基准数的右边界排好序
}
}
解释一下,由于递归过程中数组的长度是不断发生变化的,所以用L
和 R
来表示当前在操作数组区域,partition
的作用是返回数组中与基准数相同的区间。
public static int[] partition(int[] a, int p, int right, int num) {
int left = p;
while (p < right) {
if (a[p] < num) {
swap(a, left++, p++);
} else if (a[p] > num) {
swap(a, p, --right);
} else {
p++;
}
}
return new int[] {left, right};
}
实际上跟荷兰国旗的代码也就最后加了一句返回值(长度为2的数组)。
分析一下,上面的这个快排的实现怎么就比经典快排要快呢?
很明显, 上述的快排的实现方式它可以将相等的区域直接排好序,而经典快排每次递归只将一个数字排好序。这就大大地提高了运行效率,想象一下,如果数组中的数字全都为同一个,那么上述的快排的时间复杂度仅为O(n)!
最后:面试要是考快速排序,用上它之后就等着收offer吧!