C++算法之快速排序

快速排序

快速排序

• 快速排序是一种高效的排序算法，其主要思想是基于分治思想，通过确定一个分界点，将待排序的序列划分为两部分，左边部分小于等于分界点，右边部分大于等于分界点，然后对两部分分别递归地进行排序
• 快速排序的平均时间复杂度为O(nlogn)，其中n为待排序序列的长度，空间复杂度取决于递归的深度，最坏情况下为O(n)

步骤

1. 确定分界点

   • 分界点的选择可以是数组的第一个元素、中间元素、最后一个元素或者随机选择一个元素。

2. 重新划分区间

   • 将待排序序列按照分界点的值重新排列，使得左边部分小于等于分界点，右边部分大于等于分界点。这一步可以使用暴力求解法或者指针法：

   • 暴力求解法

     • 开辟两个数组a[ ]和b[ ]，对待排序序列进行扫描，将小于等于分界点的元素插入数组a[ ]，将大于等于分界点的元素插入数组b[ ]，然后将数组a[ ]和数组b[ ]进行合并。

   • 指针法

     • 使用两个指针i 和 j ，分别指向序列的起始和末尾，逐步向中间移动，直到 i 和 j 相遇。在移动的过程中，交换元素使得小于等于分界点的元素位于左边，大于等于分界点的元素位于右边。

3. 递归处理左右两段

   • 对分界点左右两边的子序列分别进行递归排序

例题

给定你一个长度为 n 的整数数列。
请你使用快速排序对这个数列按照从小到大进行排序。
并将排好序的数列按顺序输出。

输入格式
输入共两行，第一行包含整数 n。
第二行包含 n 个整数（所有整数均在 1∼109 范围内），表示整个数列。

输出格式
输出共一行，包含 n 个整数，表示排好序的数列。

数据范围
1 ≤ n ≤ 100000

输入样例：
5
3 1 2 4 5

输出样例：
1 2 3 4 5

#include<iostream>
using namespace std;

const int N = 1e6 + 10;
int n;
int q[N];

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

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

如果没有swap( )函数可以使用

if(i < j)
{
    int t = q[i];
    q[i] = q[j];
    q[j] = t;
}

#include<iostream>
using namespace std;

const int N = 1e6 + 10;
int n;
int q[N];

void quick_sort(int q[] , int l , int r)
{
    if(l >= r)
        return;
    
    int x = q[(l + r + 1) / 2];
    int i = l - 1;
    int 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(q , l , i - 1);
    quick_sort(q , i , r);
}

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

注意上面的quick_sort( )函数是不一样的，在第二个函数里 int x = q[(l + r + 1) / 2] 还可以写成 int x = q[r] ;但是绝对不能写成 int x = q[l] ; 否则会造成边界问题。同理右边届也存在这种问题

快速选择

例题

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

输入格式
第一行包含两个整数 n 和 k。
第二行包含 n 个整数（所有整数均在 1∼109 范围内），表示整数数列。

输出格式
输出一个整数，表示数列的第 k 小数。

数据范围
1 ≤ n ≤ 100000,
1 ≤ k ≤ n

输入样例：
5 3
2 4 1 5 3
    
输出样例：
3

#include<iostream>
using namespace std;

const int N = 1e6 + 10;
int n , k;
int q[N];

int quick_sort(int l , int r, int k)
{
    if(l == r)
        return q[l];
    
    int x = q[l];
    int i = l - 1;
    int j = r + 1;
    
    while(i < j)
    {
        while(q[++i] < x);
        while(q[--j] > x);
        if(i < j)
            swap(q[i] , q[j]);
    }
    
    int s1 = j - l + 1;
    if(k <= s1)
        return quick_sort(l , j, k);
    else
        return quick_sort(j + 1, r , k - s1);
}

int main(void)
{
    cin >> n >> k;
    
   for(int i = 0; i < n; i++)
       scanf("%d" , &q[i]);
    
    cout << quick_sort(0 , n - 1, k);
    
    return 0;
}

归并排序

• 归并排序的主要思想：分治

• 其主要思想就是将待排序的数组分成两个子数组，然后对这两个子数组分别进行排序，最后将已排序的子数组合并成一个有序的数组

步骤

• 分解
  • 将待排序的数组分成两个子数组，通常选择中间点作为分解点，即将数组从中间分成左右两部分
• 递归排序
  • 对左右两个子数组分别进行递归排序，直到子数组长度为1（即只有一个元素），此时认为已经排好序
• 归并
  • 将两个已排序的子数组合并成一个有序的数组。合并过程中，比较两个子数组的第一个元素，将较小（或较大）的元素放入临时数组，并向后移动指针，直到其中一个子数组的所有元素都被合并到临时数组中。最后，将剩余的子数组中的元素依次放入临时数组中，得到最终的有序数组
• 这个过程一直递归进行，直到整个数组都被排序完成。归并排序的时间复杂度为 O(nlogn)，其中 n 是待排序数组的长度

例题

给定你一个长度为 n 的整数数列。
请你使用归并排序对这个数列按照从小到大进行排序。
并将排好序的数列按顺序输出。

输入格式
输入共两行，第一行包含整数 n。
第二行包含 n 个整数（所有整数均在 1∼109 范围内），表示整个数列。

输出格式
输出共一行，包含 n 个整数，表示排好序的数列。

数据范围
1 ≤ n ≤ 100000

输入样例：
5
3 1 2 4 5

输出样例：
1 2 3 4 5

#include<iostream>
using namespace std;

const int N = 1e6 + 10;

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

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 i = l;
    int j = mid + 1;
    int k = 0;
    
    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(void)
{
    scanf("%d" , &n);
    for(int i = 0; i < n; i++)
        scanf("%d" , &q[i]);
    
    merge_sort(q , 0 , n - 1);
    
    for(int i = 0; i < n; i++)
        printf("%d " , q[i]);
    
    
    return 0;
}

例题

给定一个长度为 n 的整数数列，请你计算数列中的逆序对的数量。
逆序对的定义如下：对于数列的第 i 个和第 j 个元素，如果满足 i<j 且 a[i]>a[j]，则其为一个逆序对；否则不是。

输入格式
第一行包含整数 n，表示数列的长度。
第二行包含 n 个整数，表示整个数列。

输出格式
输出一个整数，表示逆序对的个数。

数据范围
1 ≤ n ≤ 100000，
数列中的元素的取值范围 [1,109]。

输入样例：
6
2 3 4 5 6 1

输出样例：
5

时间复杂度：该方法的时间复杂度为 O(n log n)

#include<iostream>
using namespace std;

typedef long long ll;

const int N = 1e6 + 10;
int n;
int q[N] , tmp[N];

ll merge_sort(int l , int r)
{
    if(l >= r)
        return 0;
   	
    int mid = l + r >> 1;
    ll res = merge_sort(l , mid) + merge_sort(mid + 1 , r);
    
    int k = 0;
    int i = l;
    int j = mid + 1;
    
    while(i <= mid && j <= r)
        if(q[i] <= q[j])
            tmp[k++] = q[i++];
    	else
        {
            tmp[k++] = q[j++];
            res += mid - i + 1;
        }
    
    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];
    
    return res;
}

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