这道题用传统快排(如下所示)的结果就是最后三个点TLE:
void swap(int &a, int &b)
{
int tmp = a;
a = b;
b = tmp;
}
void quickSort(int a[], int left, int right)
{
if (left >= right)
return;
int i = left, j = right;
while (i < j)
{
while (j > i && a[j] >= a[left])
j--;
while (i < j && a[i] <= a[left])
i++;
swap(a[i], (i == j) ? a[left] : a[j]); //i和j相遇则与枢轴元素交换,否则a[i]与a[j]交换
}
quickSort(a, left, i-1);
quickSort(a, j+1, right);
}
因为快排对于一些特殊的情况(例如序列原本就有序、有大量重复元素等等)会进行很多完全不必要的操作,耗费大量时间。为此,我们基于上述普通快速排序算法一步步进行优化。
一、随机化
如果永远取第一个元素作为枢轴的话,在数组已经有序的情况下每次划分都将得到最坏的结果,时间复杂度退化为O(n^2)。因为其中一个子序列每次都只比原序列少一个元素,该侧的递归深度将达到最大。
然而,我们可以通过随机选取枢轴元素来打破这种固定模式,这样每次都是最坏划分的概率就非常小了。实现起来只需要先将随机选中的元素和第一个元素交换一下位置作为枢轴元素,然后就可以接着用原来的方法进行排序了。
void quickSort(int a[], int left, int right)
{
if (left >= right)
return;
**int i = left, j = right, pivot = rand() % (right - left + 1) + left;**
**swap(a[left], a[pivot]);**
while (i < j)
{
while (j > i && a[j] >= a[left])
j--;
while (i < j && a[i] <= a[left])
i++;
swap(a[i], (i == j) ? a[left] : a[j]);
}
quickSort(a, left, i-1);
quickSort(a, j+1, right);
}
二、小区间插入排序
当序列长度分割到足够小后,继续使用快速排序递归分割的效率反而没有直接插入排序高。因此我们可以增加一个判断,当区间长度小于10以后改为使用插入排序。