算法思路
快速选择:
1.即任意选一个数,将数组划分为二。
2.最终根据该数所在的位置,即第?大,选择第k大的数字所在区间进行划分。
时间复杂度分析,第一次划分n,下一次划分期望n/2,n/4…累加和小于等于2n故快速选择算法是时间复杂度为O(n)的算法。
理解该算法的难点即一次划分之后,指针j值向哪一个数
即j指向左区间最后一个数,区间划分为即[l,j] [j+1,r]
当选取的x为重复值时呢?
去重,桶排
#include <iostream>
using namespace std;
int n,k;
const int N = 100010;
int q[N];
int quick_sort(int l,int r,int q[],int k)
{
//中止找到k
if (l >= r)
return q[l];
//先找个分界点x,将数组分成两边
int x = q[l+r >> 1];
int i = l-1;
int j = r+1;
//将数据分为LEFT<=x,和RIGHT>=x,
while(i<j)
{
do i++;while(q[i]<x);
do j--;while(q[j]>x);
if(i<j)
swap(q[i],q[j]);
}
//j停下的位置是哪里???
//k比LEFT大则递归LEFT,反之递归RIGHT
if(j-l+1 >= k)
return quick_sort(l,j,q,k);
else
return quick_sort(j+1,r,q,k - (j - l + 1));
}
int main()
{
cin>>n>>k;
for(int i = 0;i < n;i ++)
scanf("%d",&q[i]);
cout<<quick_sort(0,n-1,q,k)<<endl;
return 0;
}