第K个数 (快速选择排序)

题目:

代码讲解:

#include <iostream>
using namespace std;
const int N=1000010;
int n,k;
int num[N];

int quick_sort(int num[],int l,int r,int k){
    if(l>=r) return num[l];
    int mid = num[l], i = l-1,j = r+1;//mid可以取l或l+r>>2,不能够取r,取r会出现死循环,出现超出内存的情况
    while (i<j){
        do i++; while (num[i]<mid);
        do j--;while (num[j]>mid);
        if(i<j) swap(num[i],num[j]);
    }
    //左区间所有的数是整个区间最小的一些数,j-l+1 = 左边区间的长度 ,判断左区间数的个数是否大于k,则能判断出题目所求第k小的数在左区间,相反若小于k,则可容易判断出在右区间(注意:若判断出在右区间则题目所求第k小的数在右区间相当于第k-(j-l+1)个数
    //判断出在左或右区间后,递归左或右区间,直到递归到只有一个数,return num[l] or num[r],则必定是答案,模拟结果见图一,图二
    if (j-l+1 >= k) quick_sort(num,l,j,k);
   else  quick_sort(num,j+1,r,k-(j-l+1));
}

int main()
{
    scanf("%d%d", &n, &k);

    for (int i = 0; i < n; i ++ ) scanf("%d", &num[i]);

    cout << quick_sort(num, 0, n - 1, k) << endl;

    return 0;
}

    mid=l+r>>2

                                                                      图一

 mid=l

图二