hihocoder1133 二分·二分查找之k小数

思路：

类似于快排的分治算法。

实现：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;

const int MAXN = 1000000;
int a[MAXN + 5], n, k;

int partition(int * num, int l, int r)
{
    int i = l, j = r, m = (l + r) >> 1;
    int x = num[m];
    while (i != j)
    {
        while (j > m && num[j] >= x)
        {
            j--;
        }
        num[m] = num[j], m = j;
        while (i < m && num[i] <= x)
        {
            i++;
        }
        num[m] = num[i], m = i;
    }
    num[i] = x;
    return i;
}

int findK(int * num, int l, int r, int k)
{
    int pos = partition(num, l, r);
    if (pos == k)
    {
        return num[k];
    }
    if (pos > k)
    {
        return findK(num, l, pos - 1, k);
    }
    return findK(num, pos + 1, r, k);
}

int main()
{
    scanf("%d %d", &n, &k);
    for (int i = 1; i <= n; i++)
    {
        scanf("%d", &a[i]);
    }
    printf("%d
", findK(a, 1, n, k));
    return 0;
}