#1133 : 二分·二分查找之k小数

描述

在上一回里我们知道Nettle在玩《艦これ》，Nettle的镇守府有很多船位，但船位再多也是有限的。Nettle通过捞船又出了一艘稀有的船，但是已有的N(1≤N≤1,000,000)个船位都已经有船了。所以Nettle不得不把其中一艘船拆掉来让位给新的船。Nettle思考了很久，决定随机选择一个k，然后拆掉稀有度第k小的船。 已知每一艘船都有自己的稀有度，Nettle现在把所有船的稀有度值告诉你，希望你能帮他找出目标船。

提示：非有序数组的二分查找之二

输入

第1行：2个整数N,k。N表示数组长度，
第2行：N个整数，表示a[1..N]，保证不会出现重复的数，1≤a[i]≤2,000,000,000。

输出

第1行：一个整数t，表示t在数组中是第k小的数，若K不在数组中，输出-1。

样例输入

10 4
1732 4176 2602 6176 1303 6207 3125 1 1011 6600

样例输出

1732

利用了快速排序的思想，对于它理解比较少，恐怕遇到这种思想的题，不一定能a。

代码：

#include <bits/stdc++.h>

using namespace std;
const int ma=1010000;
int a[ma];

int main(){
    int n,k;
    while(~scanf("%d%d",&n,&k))
    {
        memset(a,0,sizeof(a));
        if(k>n)
        {
            printf("-1\n");
            continue;
        }
        for(int i=0;i<n;i++)
        {
            scanf("%d",&a[i]);
        }
        int l=0,r=n-1;
        int ri,le,m;
        while(l<=r)
        {
            ri=r;
            m=a[le=l];
            while(le<ri)
            {
                while(le<ri&&a[ri]>m)ri--;
                a[le]=a[ri];
                while(le<ri&&a[le]<m)le++;
                a[ri]=a[le];
            }
            a[le]=m;
            if(le<k-1)l=le+1;
            else r=le-1;
        }
        printf("%d\n",a[r+1]);

    }

    return 0;
}