基础算法1——排序

排序算法

1、快速排序$O(nlogn)$

问题描述：对于数列a[l, r]​，按从小到大排列。

快速排序三步骤：

• 确定分界点：常选择 x = a[l]、a[(l+r)/2]、 a[r]​

• 调整范围：通过某些办法把$\le x$的数移到数组a的左边，$\ge x$的数移到数组a的右边。

• 递归处理左右两段（分治思想）

这里最关键的是怎么把当前数组a[]分成两段。注意：分成两段后，x不一定会集中在分界线附近的，这随便找个例子都能验证。介绍一种双指针法来完成这个分两段的工作：

两个指针分别从两侧向中间走，如果begin走到某个位置，这个位置的值大于x，那begin停下来；接着end开始走，走到某个位置，这个位置的值小于x，end停下来。接着交换begin/end位置的值。然后指针接着往中间走。这样一定能走到begin == end 或者 begin > end的情况。说明已经分成两段了：

/*快速排序代码模板*/
#include <iostream>
using namespace  std;

void quick_sort(int q[], int l, int r)
{
    if (l >= r) return;

    int i = l - 1, j = r + 1, x = q[l + 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(q, l, j);     
    quick_sort(q, j + 1, r);
}

int main() {
    int a[10] = {5, 2, 3, 1, 9, 7, 4, 6, 8, 0};
    int n = 10;
    quick_sort(a, 0, n - 1);
    for (int i = 0; i < n; i ++) printf("%d ", a[i]);
    return 0;
}
// output:
0 1 2 3 4 5 6 7 8 9

注意这里容易出现边界问题：

x = q[l]; 
...
quick_sort(q, l, i - 1);
quick_sort(q, i, r);
//=============================
x = q[r];
...
quick_sort(q, l, j);
quick_sort(q, j + 1, r);

比如我是上面这两种组合，去试试看a = {1, 2}这个例子，会发现最后都会死循环了。所以可以背模板那种套路就行了，省的现场在那调边界问题，毕竟思路你是知道的。

2、 归并排序$O(nlogn)$

也是三步骤：

• 确定分界点：mid = (l + r) / 2​，把数组分成两段；

• 递归排序left，right（分治思想）；

• 归并 – 把排序好的两段合并成一段：★★用双指针法可以完成这个事情

           ↓指针1
  1  3  5  7  9
  
  2  4  5  8  10
        ↑指针2
        
  res: 1 2 3 4 5
  

/*归并排序代码模板*/
#include <iostream>
using namespace  std;

void merge_sort(int q[], int l, int r)
{
    if (l >= r) return;

    int mid = (l + r) >> 1;
    merge_sort(q, l, mid);  // 对左右两段分别排序
    merge_sort(q, mid + 1, r);

    int k = 0, i = l, j = mid + 1;
    int temp[r - l + 1];
    // 双指针将两段排好序的数列进行合并
    while (i <= mid && j <= r){
        if (q[i] <= q[j]) temp[k ++] = q[i ++];
        else temp[k ++] = q[j ++];
    }

    // 两个while至多一个需要运行，否则上面的while出不来
    while (i <= mid) temp[k ++] = q[i ++];
    while (j <= r)   temp[k ++] = q[j ++];

    for (i = l, j = 0; i <= r; i ++, j ++) q[i] = temp[j];
}

int main() {
    int a[10] = {5, 2, 3, 1, 9, 7, 4, 6, 8, 0};
    int n = 10;
    merge_sort(a, 0, n - 1);
    for (int i = 0; i < n; i ++) printf("%d ", a[i]);
    return 0;
}
// output:
0 1 2 3 4 5 6 7 8 9

下一讲：基础算法2——二分