快速排序简洁写法

云归而-暝

已于 2022-03-23 22:28:58 修改

阅读量434

点赞数 3

分类专栏：算法文章标签： c++ 快速排序

于 2022-01-01 14:14:50 首次发布

本文链接：https://blog.csdn.net/qq_45256489/article/details/122267067

版权

算法专栏收录该内容

12 篇文章 0 订阅

订阅专栏

本文深入探讨了快速排序算法的实现细节，包括常见的错误点和优化策略。首先，介绍了基本的快速排序算法，强调了分区操作的边界处理和递归调用的正确方式。然后，提出了使用舍伍德算法进行随机枢轴选择以改善最坏情况下的时间复杂度。最后，针对含有大量重复元素导致性能退化的场景，给出了改进版的三向切分快速排序。整个讨论旨在提高快速排序算法的效率和稳定性。

摘要由CSDN通过智能技术生成

代码

int partition(vector<int> &a, int s, int e) {
  for (int i = s; i <= e; swap(a[s++], a[i++]))
    while (a[e] < a[i]) i++;
  return s - 1;
}
void quick_sort(vector<int> &a, int s, int e) {
  if (s >= e) return;
  int m = partition(a, s, e);
  quick_sort(a, s, m - 1);
  quick_sort(a, m + 1, e);
}

快速排序比较容易写错的点就两个

partition 里的边界判断。这里我把枢轴选在区间最后一个数，判断时又没带等号,所以while循环必然会在最后一个数字上停下。不用判断越界。i==e时顺便把最后一次交换也做了
递归调用quick_sort时一定要m-1和m+1这样可以保证长度减一，否者可能会死循环。

另一种受Linux内核源代码启发的写法

void quick_sort(vector<int> &a, int s, int e) {
  if (s >= e) return;
  int t = s;
  int m = ({
    for (int i = s; i <= e; swap(a[s++], a[i++]))
      while (a[e] < a[i]) i++;
    s - 1;
  });
  quick_sort(a, t, m - 1);
  quick_sort(a, m + 1, e);
}

优化

为了优化快速排序最坏复杂度。
用了舍伍德算法(Sherwood)随机选取枢轴。
代码如下

std::random_device rd;
int partition(vector<int> &a, int s, int e) {
  for (int i = s; i <= e; swap(a[s++], a[i++]))
    while (a[e] < a[i]) i++;
  return s - 1;
}
void quick_sort(vector<int> &a, int s, int e) {
  if (s >= e) return;
  swap(a[rd() % (e - s + 1) + s], a[e]);
  int m = partition(a, s, e);
  quick_sort(a, s, m - 1);
  quick_sort(a, m + 1, e);
}

另外提一句,取模获得某范围的随机数其实不靠谱。这里为了方便直接就取模了。
举个例子，假设原来随机数的范围是0,1,2各 $\frac13$ 的概率。如果你模2，那么0的概率就变成了 $\frac23$ ，1的概率还是 $\frac13$ ，就不是等概率了。

后记

上面的属于快慢指针法的快排。遭到洛谷的毒瘤数据暴打。如果数组中有很多相同元素。就会退化为 $O(n^2)$ 。
附上改进写法。

void quick_sort(int a[], int s, int e) {
  if (s >= e) return;
  int k = a[(s + e) / 2], i = s, j = e;
  while (i <= j) {
    while (a[i] < k) i++;
    while (a[j] > k) j--;
    if (i <= j) swap(a[i++], a[j--]);
  }
  quick_sort(a, s, j);
  quick_sort(a, i, e);
}

有以下注意点

退出循环时要保证i>j,否则当i==j时可能会死循环。比如i=j=s时,会递归调用quick_sort(i, e);即quick_sort(s, e);。
swap前的 if (i <= j)不能少，否则当i>j时会导致乱序。
if (i <= j)也不能改成if (i < j)，否则i==j时会死循环。
内部while循环条件不能带等号，否则会越界。