[洛谷 P1177] 【模板】快速排序

题目描述

利用快速排序算法将读入的 NN 个数从小到大排序后输出。

快速排序是信息学竞赛的必备算法之一。对于快速排序不是很了解的同学可以自行上网查询相关资料，掌握后独立完成。（C++C++ 选手请不要试图使用 STL，虽然你可以使用 sort 一遍过，但是你并没有掌握快速排序算法的精髓。）

输入格式

第 11 行为一个正整数 NN，第 22 行包含 NN 个空格隔开的正整数 a_iai​，为你需要进行排序的数，数据保证了 A_iAi​ 不超过 10^9109。

输出格式

将给定的 NN 个数从小到大输出，数之间空格隔开，行末换行且无空格。

输入输出样例

输入 #1复制

5
4 2 4 5 1

输出 #1复制

1 2 4 4 5

说明/提示

对于 20\%20% 的数据，有 N\leq 10^3N≤103；

对于 100\%100% 的数据，有 N\leq 10^5N≤105。

AC:

#define _CRT_SECURE_NO_WARNINGS
#include <cstdio>

int a[100005];

template <typename T>
void swap(T& a, T& b)
{
    T tmp = a; a = b; b = tmp;
}


template <typename T>
void insert_sort(T a[], int left, int right)
{
    for (int i = left+1; i <= right; ++i)  // 依次拿牌
        for (int j = i ; j - 1 >= left && a[j] < a[j-1]; --j) // 每张牌尝试向前移动   
            swap(a[j-1], a[j]);  
}

template <typename T>
void sort(T a[], int left, int right)
{
    if (left >= right) return;
    if (right - left < 20)  // 数据少于20个的时候采用插入排序
    {
        insert_sort(a, left, right);
        return;
    }
    bool _is_ordered = true;
    for (int i = left + 1; i <= right; ++i) if (a[i] < a[i - 1]) _is_ordered = false;
    if (_is_ordered) return;  // 本身有序的数据不排序
    int l = left, r = right, pivot = (l + r) >> 1;
    swap(a[pivot], a[l++]);  // 排序时选取pivot之后，将pivot放到数组最低的位置（left）, 同时将左边的指针向后移动一位
    while (l <= r)
    {
        while (a[l] <= a[left] && l < right) ++l; // 找比 pivot 大的值，最大不能超过右边边界
        while (a[r] >= a[left] && r > left) --r;  // 找比 pivot 小的值，最小不能超过左边边界
        if (l >= r) break; // 能交换的都已经交换结束了，可以退出交换的步骤
        swap(a[l], a[r]);  // l < r 成立，依然可以交换
    }
    // 退出上面的循环的时候， i >= L 的数据都满足 a[i] < a[left], i <= R 的数据都满足 a[i] < a[left]
    swap(a[left], a[r]);  // 因为 a[left] 放在左边，所以要和比其小的数据进行交换，即和 a[r] 进行交换
    if (r > left) sort(a, left, r - 1);
    if (r < right) sort(a, r + 1, right);
}

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

分析：

while 有两个作用

1. 在 l < r 的时候进行 swap(a[l], a[r])

2. 保证完成了这个 while 流程之后，swap(a[left], a[r]) 是正确的，也就是在完成了这个 while 流程之后，应该保证:

 

对于 任意的 i, 满足 left + 1<= i <= r， 有 a[left] < a[i]。

对于 任意的 i, 满足 r <= i <= right，有 a[right] > a[i]。

 

为什么 while(l < r) 不对

如果写成 while(l < r)，则考虑简单的情况 ： a[] = {1, 2}, left = 0, right = 1;

假设编译器是取下整的，则 pivot = lower(0+1) >> 1 = 0, 然后 swap(a[left], a[pivot]) 得到的结果是

swap(a[0], a[0]), a[] = {1, 2} 不变。

 

接着 ++l ，即将左边的指针向右移动一位。此时 l = r = 1，即左右两个指针重合了。

然后尝试进入外层的 while 循环：while (l < r)，进不去因为此时 l == r && l == 1。

此时能否保证上面的第二个作用生效呢？答案是不能。因为没有交换，也就是没有办法保证 <= r 的所有数据都满足 小于开始的 a[pivot]，那么后面再进行 swap(a[left], a[r]) 的时候，就发生了 :

swap(a[left], a[r]) = swap(a[0], a[1])，结果是 a[] = {2, 1} 数据从有有序变成了无序。

 

但是如果写成是 while (1) 或者是 while (l <= r) 就可以避免这个问题，即使区间大小为2，此时移动完 pivot 到了左边之后，仍然可以进入 while 循环，将 l 和 r 指针移动到正确的位置。看看写成 while (1) 或者 while (l <= r) 之后发生的情况：

 

a[] = {1, 2}

 

依然假设编译器自动向下取整，首先 pivot = (l + r) >> 1 = 0，swap(a[pivot], a[l++])，此时就会发生

swap(a[0], a[0]) ，a[] = {1, 2}，之后得到的情况为：a[] = {1, 2}， l == r && l == 1，然后进入了 while 循环，在这个循环里，首先移动 l , r 指针到正确的位置上，左边的指针 l 无法移动，已经到了最右边，但是右边的指针还可以移动，右边的指针会发现此时 2 >= 1，继续左移，然后来到最左边，r == 0，之后判断 l >= r 成立，break，然后 swap(a[l], a[r])，swap(a[0], a[0])，a[] = {1, 2} 没有问题，算法结束。

 

和STL::SORT对比

 

手写的：

 

STL：

第三组数据比标准库慢了很多，其他的都达到了标准库的水平。