[排序] 快速排序、归并排序模板(模板)

0. 前言

正经人谁手写这玩意？

1. 快速排序

785. 快速排序

本质： 分治。是不稳定的排序算法。

时间复杂度：$O(nlogn)$
空间复杂度：$O(logn)$，基于递归，用到栈空间

模板注意点：

• 注意点就是不要随便改动里面的任何一个下标…
• 快排边界情况很多很多，这是个经过千锤百炼的模板代码，理解算法思想后背过模板就行了
  • 基准值 int x = q[l+r>>1]，写成 int x = q[l] 也是可以的，但是很容易被卡…写成 int x = q[l+r>>1] 代表除 2，基准值取左中位数，或是写成 int x = q[l] 的时候，就一定在递归的时候 quick_sort(l, j); quick_sort(j + 1, r); 一定注意是 j 和 j + 1
  • 举个简单的边界情况，因为是 j = i + 1，所以当 int x = q[l], i = l - 1, j = r + 1; 时，将递归部分修改为 quick_sort(l, i - 1); quick_sort(i, r); 看似是等价的修改，完全就是等价代换么…但是针对 [1,2] 这个数组，直接 MLE。因为一开始 l = 0, r = 1 开始 quick_sort，基准值取 x=q[0] = 1，i = -1, j = 2; 进入 while 后 do while 先加一次，while 判断失败，i 结束，此时 i=0。j-- 先来一次，然后while 判断成功，j-- 再次执行，while 判断失败。两次 do while 结束，进入 if 判断，此时 i=j=0 不成立。quick_sort(l, i - 1); 等价于 quick_sort(0, -1); 会自动结束。然而 quick_sort(i, r); 等价于 quick_sort(0, 1);，这不和一开始传入 l = 0, r = 1 一样吗？就造成了无限递归…MLE 了
  • 那么，如果在此传入的是 quick_sort(l, j); 就等价于 quick_sort(0, 0); 会自动结束，quick_sort(j+1, r); 就等价于 quick_sort(1, 1); 也会自动结束。所以能够正常结束。故当基准值写成 int x = q[l] 的时候，递归不要写成 quick_sort(l, i-1);quick_sort(i,r);
  • 同理，当基准值写成 int x = q[r] 的时候，递归不要写成 quick_sort(l, j);quick_sort(j+1, r); 相同例子，无限递归…

关于无限递归问题，大家动手模拟一遍样例就能理解了。快排边界问题很多，很烦…也有快排优化的版本，如单边递归、三数取中等等，以及将快排修改为稳定排序的方法都很多。有兴趣可自行研究。

i 和 j 可能相等，可能穿越。

模板代码：

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 1e5+5;

int n;
int q[N];

void quick_sort(int l, int r) {
    if (l >= r) return ;
    int x = q[l+r>>1], i = l - 1, j = r + 1;
    while (i < j) {
        do i++; while (q[i] < x);
        do j--; while (q[j] > x);
        if (i < j) swap(q[i], q[j]);
    }
    quick_sort(l, j);	// j对应 q[l]
    quick_sort(j + 1, r);
}

int main() {
    cin >> n;
    for (int i = 0; i < n; ++i) cin >> q[i];
    
    quick_sort(0, n - 1);
    
    for (int i = 0; i < n; ++i) cout << q[i] << ' ';
    
    return 0;
}

2. 归并排序

787. 归并排序

本质： 分治。是稳定的排序算法。

时间复杂度：$O(nlogn)$
空间复杂度：$O(n)$，辅助数组空间，完成二路归并

模板注意点：

• 手别残，理解二路归并思想很好写
• 建议模拟样例一般，啥都懂了

模板代码：

#include <iostream>

using namespace std;

const int N = 1e5+5;

int n;
int q[N], tmp[N];

void merge_sort(int l, int r) {
    if (l >= r) return ;
    int mid = l + r >> 1;
    merge_sort(l, mid), merge_sort(mid + 1, r);
    int k = 0, i = l, j = mid + 1;
    while (i <= mid && j <= r) {
        if (q[i] <= q[j]) tmp[k++] = q[i++];
        else tmp[k++] = q[j++];
    }
    while (i <= mid) tmp[k++] = q[i++];
    while (j <= r) tmp[k++] = q[j++];
    for (int i = l, j = 0; i <= r; ++i, ++j) q[i] = tmp[j];
}

int main() {
    cin >> n;
    for (int i = 0; i < n; ++i) cin >> q[i];
    
    merge_sort(0, n - 1);
    
    for (int i = 0; i < n; ++i) cout << q[i] << ' ';
    
    return 0;
}