快速排序与快速选择

快速排序

核心思路

快速排序的核心思路是分治，不断选择分界点以使分界点左侧的数小于x，分界点右侧的数大于x，并进行不断递归，直至完成排序。该算法的时间复杂度是O(n*logn)。

实现方法

使用双指针方法可以实现快速排序，flagi与flagj从数组两侧开始遍历，发现第一个不满足要求的数字时停止，并将i、j互相交换。要注意边界问题，避免陷入死循环，快排模板如下。

#include<iostream>
#include<cstdio>
using namespace std;
const int N=1e6+7;
int n;
int q[N];
void quick_sort(int q[],int l,int r)
{
    if(l>=r) return;
    int x=q[l+r>>1],i=l-1,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()
{
    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;
}

求解数组中第k小的数（快速选择算法）

只选择一侧区间递归，若在左侧区间，则是左侧区间的第k小个数，若在右侧区间，则是右侧区间的第k-sl小的数，sl为此时左侧区间的长度。

#include<iostream>
using namespace std;
const int N=1e5+9;
int n,k;
int q[N];
void quick_sort(int k,int l,int r)
{
    if(l>=r) return;
    int x=q[l+r>>1],i=l-1,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]);
    }
    int sl=j-l+1;
    if(sl>=k) quick_sort(k,l,j);
    else quick_sort(k-sl,j+1,r);
}
int main()
{
    cin>>n>>k;
    for(int i=0;i<n;i++) cin>>q[i];
    quick_sort(k,0,n-1);
    cout<<q[k-1];
    return 0;
}