快速排序的二分优化法

普通的快速排序代码：

int partition1(int a[],int left,int right){
    int p=(int)round(1.0*rand()/RAND_MAX*(right-left)+left);
    swap(a[p],a[left]);
    int temp=a[left];
    while(left<right){
        while(left<right&&a[right]>temp) right--;
        a[left]=a[right];
        while(left<right&&a[left]<=temp) left++;
        a[right]=a[left];
    }
    a[left]=temp;
    return left;
}
void quicksort(int a[],int left,int right){
    if(left<right){
        int p=partition1(a,left,right);
        quicksort(a,left,p-1);
        quicksort(a,p+1,right);
    }
}

利用二分法，可以优化该代码，加快速度。

void qsort(int l,int r)//应用二分思想
{
    int mid=a[(l+r)/2];//中间数
    int i=l,j=r;
    do{
        while(a[i]<mid) i++;//查找左半部分比中间数大的数
        while(a[j]>mid) j--;//查找右半部分比中间数小的数
        if(i<=j)//如果有一组不满足排序条件（左小右大）的数
        {
            swap(a[i],a[j]);//交换
            i++;
            j--;
        }
    }while(i<=j);//这里注意要有=
    if(l<j) qsort(l,j);//递归搜索左半部分
    if(i<r) qsort(i,r);//递归搜索右半部分
}

这样优化后的速度会快一点。
另外，这种优化算法对应的随机选择算法也更容易写，因为在最后其他条件的情况就是选择的对象。（具体是将区间分为三个区间，可以看以下代码分析）

#include<bits/stdc++.h>
using namespace std;
int x[5000005],k;
void qsort(int l,int r)
{
	int i=l,j=r,mid=x[(l+r)/2];
	do
	{
		while(x[j]>mid)
			j--;
		while(x[i]<mid)
			i++;
		if(i<=j)
		{
			swap(x[i],x[j]);
			i++;
			j--;
		}
	}
	while(i<=j);
	//快排后数组被划分为三块： l<=j<=i<=r
	if(k<=j) qsort(l,j);//在左区间只需要搜左区间
	else if(i<=k) qsort(i,r);//在右区间只需要搜右区间
	else //如果在中间区间直接输出
	{
		printf("%d",x[j+1]);
		exit(0);
	}
}
int main()
{
	int n;
	scanf("%d%d",&n,&k);
	for(int i=0;i<n;i++)
		scanf("%d",&x[i]);
	qsort(0,n-1);
}

可以对比先前的随机选择算法。这两种随机选择算法复杂度都是o(n)。

#include <iostream>
#include <algorithm>
#include <ctime>
#include <cstdlib>

using namespace std;
static int a[50000100];
int partition1(int a[],int left,int right){
    int p=(int)(1.0*rand()/RAND_MAX*(right-left)+left);
    swap(a[p],a[left]);
    int temp=a[left];
    while(left<right){
        while(left<right&&a[right]>temp) right--;
        a[left]=a[right];
        while(left<right&&a[left]<=temp) left++;
        a[right]=a[left];
    }
    a[left]=temp;
    return left;
}
int partitionselect1(int a[],int left,int right,int m){
    if(left==right) return a[left];
    int p=partition1(a,left,right);
    int M=p-left+1;
    if(m<M) return partitionselect1(a,left,p-1,m);
    else if(m>M) return partitionselect1(a,p+1,right,m-M);
    else return a[p];
}
int main()
{
    int n,m;
    srand((unsigned)time(NULL));
    //cin>>n>>m;
    scanf("%d %d",&n,&m);
    for(int i=0;i<n;i++){
        //cin>>a[i];
        scanf("%d",&a[i]);
    }
    //cout<<partitionselect1(a,0,n-1,m);
    printf("%d",partitionselect1(a,0,n-1,m+1));
    return 0;
}```