求无序数组的第K(大/小)数的三种方法

题目描述：给定一个无序数组（长度为n）和一个正整数 K，让你求此数组中第K大的数

方法1：最简单的思路，你肯定会想到 先把数组排序，然后再取下标为K的对应数组元素。这种方法肯定是可行的，但是这种方法无端的对咱们不需要查找的元素也进行了处理，在一定程度上增加了算法的消耗！这样是不值得的！此方法的平均时间复杂度为O(n*logn).

方法2：利用堆调整算法的性质，建立一个大小为K的堆，然后将整个数组元素添加到其中，利用其自动调整的性质，不断移除最小的元素，最后取堆顶的元素，这个元素就是我们需要求得的第K大元素。可以用优先队列去进行实现。这个方法其实来说也是一种相对不错的方法。因为一个堆的元素数目为K时，堆的高度至多为logk。建立堆的时间为k*logk。而调整的次数为n-k，每次调整的时间为logk，所以调整的时间为（n-k）*logk，所以总的时间复杂度为n*logk。

#include <bits/stdc++.h>
using namespace std;
const int N = 10101;
int a[N];
int n;
int main()
{
    int k;
    while(scanf("%d %d",&n,&k)!=EOF)
    {
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
        }
        std::priority_queue<int,vector<int>,less<int> >q;
        for(int i=1;i<=n;i++)
        {
             q.push(a[i]);

             if(q.size()>k) q.pop();
        }
        cout<<q.top()<<endl;
    }
    return 0;
}

方法3：利用快速排序分治的思想，每次根据枢轴元素的对数组进行划分，并根据胡枢轴元素的位置和K的大小，决定向枢轴元素左侧或者右侧元素进行进一步划分。这样最坏时间复杂度为O(K*N)。平均时间复杂度为O(K*logK)。代码如下：

#include <bits/stdc++.h>
using namespace std;
const int N = 10101;
int a[N];
int n;
int Check_KthNumber(int start,int endd)
{
    int l,r,key;
    l = start;
    r = endd;
    key = a[start];
    while(l<r)
    {
        while(l<r&&a[r]>=key) r--;
        a[l] = a[r];
        while(l<r&&a[l]<=key) l++;
        a[r] = a[l];
    }
    a[l] = key;
    return l;
}
void GetReasult(int start,int endd,int K)
{
    if(start==endd) return ;
    else {
        int num;
        if (start < endd)
            num = Check_KthNumber(start, endd);
        if (num == K) return;
        if (num > K) {
            GetReasult(start, num - 1, K);
        } else {
            GetReasult(num + 1, endd, K);
        }
    }
}
int main()
{
    int K;
    while(scanf("%d %d",&n,&K)!=EOF)
    {
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
        }
        GetReasult(1,n,K);
        printf("%d",a[K]);
    }
    return 0;
}