灵活运用——快排

 

 

单纯用sort会超时，需要思考在快排中进行什么优化。

因为快排就是将区间划分为：（left<= j==i <=right)

　　1.小于等于x的区间

　　2.大于等于x的区间

而我们要找的是第k小的数，如果k在左区间内，则继续对左区间进行快排。如果k在右区间内，对右区间进行快排。如果k==j，则输出a[j]

 

#include <bits/stdc++.h>
using namespace std;

const int N=5e6+10;
int a[N];
int k;
void quick_sort(int l,int r)
{
    if(l>=r)return ;
    int i=l-1,j=r+1,x=a[l];
    while(i<j)
    {
        do i++;while(a[i]<x);
        do j--;while(a[j]>x);
        if(i<=j)swap(a[i],a[j]);
    }
    
    //区间分成 left<= j=i <=right 
    if(k==j)    
    {
        printf("%d",a[j]);
        return ;
    }
    else if(k<j)quick_sort(l,j);    //如果k在j左边，对左边进行快排 
    else if(k>j)quick_sort(j+1,r);    //如果k在j右边，对右边进行快排 
}
int main()
{
    int n;
    scanf("%d%d",&n,&k);
    for(int i=0;i<n;i++)scanf("%d",&a[i]);
    quick_sort(0,n-1);
    return 0;
}