题目描述:给定一个无序数组(长度为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;
}