【排序】快速排序和归并排序附加应用算法题

实现快速排序

快速排序是先利用哨兵分割两边，然后进行递归处理。

#include <iostream>
using namespace std;
const int N = 100010;
int f[N];
int n;

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

快速排序实战（第K个数）

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

#include <iostream>
#include <algorithm>
using namespace std;
const int N = 100010;
int s[N];

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 n,k;
    cin >> n >> k;
    for(int i=0; i<n;i++) cin >> s[i];
    quick_sort(s,0 ,n-1);
    cout << s[k-1];   // 直接找第k个数
    return 0;
}

实现归并排序

先分支， 后合并。这里是二路合并。

#include <iostream> 
using namespace std;

const int N = 100010;
// q是代表数组， tmp代表临时数组，归并排序中用来存放两个子数组插入的值
int q[N], tmp[N];
int 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); ///< 直接递归，分支进行排序

    // 这里的i是左边的值，j 是右边的值，写成固定的值就不能进行下一次递归了
    int i= l, j = mid + 1, k = 0;
    while(i <= mid && j<=r)   ///< 双指针进行数组的排序， 每次将较小的数组中的元素放入tmp数组中
    {
        if(q[i] <= q[j] ) tmp[k++] = q[i++];
        else    tmp[k++] = q[j++];
    }
    while(i <= mid) tmp[k++] = q[i++];   ///< 扫尾工作， 如果一个数组已经完成元素进入tmp数组，另一个数组中还有元素，那么直接放入tmp数组中
    while(j <= r) tmp[k++] = q[j++];

    ///< 物归原主， 将tmp数组存放的两个排序数组的元素放到原数组中
    for(int i=l, j=0; i<=r;j++,i++) q[i] = tmp[j];  ///< i数原数组中idx， j是tmp数组中的idx
}

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

归并排序实战（逆序对的数量）牛客网题目

给定一个长度为n的整数数列，请你计算数列中的逆序对的数量。

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

#include <iostream>
using namespace std;

const int N = 100010;
int q[N], tmp[N];
int n;
typedef long long LL;

LL merge_sort(int l, int r)
{
    if(l>=r) return 0;
    int mid = l + r >> 1 ;
    LL re = merge_sort(l,mid) + merge_sort(mid + 1, r);

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


int main()
{
    cin >> n;
    for(int i=0; i<n; i++) cin >> q[i];

    cout << merge_sort(0, n -1);
    return 0;
}