快排与归并排序

基础排序之快排与归并排序

---

快速排序

快排属于分治算法，分治算法都有三步：

1. 分成子问题
2. 递归处理子问题
3. 子问题合并

主要步骤：

1. 确定分界点，可以任选 a[l]，a[r]，a[(l + r) / 2] 其中一个作为分界点。
2. 设置两个头尾指针 i, j，初始化 i = l - 1, j = r + 1 (避免发生边界问题导致死循环) ，向中间移动。每次循环都先将 i 右移和 j 左移，然后判断，如果 a[i] < a[j] 且 i < j，就交换 a[i] 和 a[j].
3. 最后根据分界点分别递归左右两部分。

快排的一般写法如下：

void quick_sort(int q[], int l, int r)
{
    //递归的终止情况
    if(l >= r) return;
    //第一步：分成子问题
    int i = l - 1, j = r + 1, x = q[(l + r) / 2];
    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(q, l, j);
    quick_sort(q, j + 1, r);
    //第三步：子问题合并.快排这一步不需要操作，但归并排序的核心在这一步骤
}

下面看两道例题

【例一】给定一个长度为 n 的整数序列，请使用快速排序对这个序列从小到大排序后按顺序输出。(1 <= n <= 100000)

#include <iostream>
#include <algorithm>
using namespace std;
const int N = 1e5 + 10;

int n;
int a[N];

void quick_sort(int a[], int l, int r)
{
    if (l >= r) return;
    
    int i = l - 1, j = r + 1;
    int mid = a[(l + r) >> 1]; //右移一位等同于除以2
    while (i < j)
    {
        //每次循环都是先移动指针，再进行比较
        do i++; while (a[i] < mid);
        do j--; while (a[j] > mid);
        if (i < j) swap(a[i], a[j]);
    }
    
    quick_sort(a, l, j);
    quick_sort(a, j + 1, r);
}

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

【例二】第 k 个数

给定一个长度为 n 的整数数列，以及一个整数 k，请用快速选择排序算法求出数列从小到大排序后的第 k 个数。(1 <= n <= 100000, 1 <= k <= n)

#include <iostream>
#include <algorithm>
using namespace std;
const int N = 1e5 + 10;

int n, k;
int a[N];

int quick_sort(int l, int r, int k)
{
    //最后l和r都指向同一个数，随便输出一个就好了
    if (l == r) return a[l];
    
    int i = l - 1, j = r + 1;
    int mid = a[(l + r) >> 1];
    while (i < j)
    {
        while (a[++i] < mid);
        while (a[--j] > mid);
        if (i < j) swap(a[i], a[j]);
    }
    
    //以分界点为中心分成两部分，判断第k个数在哪一部分，就递归循环哪一部分
    if (k <= j - l + 1)
        return quick_sort(l, j, k);
    return quick_sort(j + 1, r, k - (j - l + 1));
}

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

---

归并排序

运用双指针的思想，先递归再合并。

主要步骤：

1. 确定分界点。与快排不同，归并每次都将中点作为分界点，将整个序列均分为两部分。mid = (l + r) / 2
2. 递归排序。对两个子序列分别设置个指针 i, j，从头开始遍历，每次比较 a[i] 和 a[j]，将小的放入一个临时序列 temp[] 中。如果有一部分遍历完了，而另一部分还有剩余，则将剩余那一部分直接接在临时序列的后面，因为剩下的这部分一定是大于前面的。
3. 归并，合二为一。将临时序列放入原序列中。

归并的一般写法如下：

void merge_sort(int a[], int l, int r)
{
    if (l >= r) return;
    
    //确定中间分界点
    int mid = (l + r) >> 1;
    
    //两边递归
    merge_sort(a, l, mid);
    merge_sort(a, mid + 1, r);
        
    int k = 0;
    int i = l, j = mid + 1;
    //比较左右两半边
    while (i <= mid && j <= r)
        if (a[i] < a[j]) temp[k++] = a[i++];
        else temp[k++] = a[j++];
    
    while (i <= mid)//左半边剩下的
        temp[k++] = a[i++];
    
    while (j <= r)//右半边剩下的
        temp[k++] = a[j++];
        
    //合并区间
    for (i = l, j = 0; i <= r; i++, j++)
        a[i] = temp[j];
}

下面看两道例题

【例一】给定一个长度为 n 的整数数列，请使用归并排序对这个数列按照从小到大排序后按顺序输出。(1 <= n <= 100000)

#include <iostream>
using namespace std;
const int N = 1e5 + 5;

int n;
int a[N], temp[N];

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

int main(){
    scanf("%d", &n);
    
    for (int i = 0; i < n; i++)
        scanf("%d", &a[i]);
    
    merge_sort(a, 0, n - 1);
    
    for (int i = 0; i < n; i++)
        printf("%d ", a[i]);
    
    return 0;
}

【例二】逆序对的数量

逆序对的定义：对于数列的第 i 个和第 j 个元素，如果满足 i < j 且 a[i] > a[j]，则称其为一个逆序对，否则不是。

简单来说，逆序对就是前面的数比后面的大。

给定一个长度为 n 的整数数列，请计算数列中的逆序对的数量。(1 <= n <= 100000, 1 <= a[i] <= 10⁹)

#include <iostream>
#define ll long long
using namespace std;
const int N = 100010;

int n;
int a[N], temp[N];

ll merge_sort(int l, int r){
    if (l >= r)
        return 0;
    
    //分成两半，分别递归 
    int mid = l + r >> 1;
    ll ans = merge_sort(l, mid) + merge_sort(mid + 1, r);
    
    int k = 0; 
    int i = l, j = mid + 1;
    while (i <= mid && j <= r){
        if (a[i] <= a[j])
            temp[k++] = a[i++];
        else
        {
            temp[k++] = a[j++];
            //计算有多少个逆序对
            ans += mid - i + 1;
        }
    }
    
    //处理剩下的
    while (i <= mid) temp[k++] = a[i++];
    while (j <= r) temp[k++] = a[j++];
    
    //合并区间
    for (int i = l, j = 0; i <= r; i++, j++)
        a[i] = temp[j];
        
    return ans;
}

int main(){
    cin >> n;
    for (int i = 1; i <= n; i++)
        scanf("%d", &a[i]);
        
    cout << merge_sort(1, n) << endl;
    
    return 0;
}

---