题目:找出无序数组中第Kth大的数,如{63,45,33,21},第2大的数45。
输入:
第一行输入无序数组,第二行输入K值。
该是内推滴滴打车时(2017.8.26)的第二题,也是《剑指offer》上最小k个数的变形。当时一看到题,这个不是用快排吗?然后就写了,结果始终没有通过,遗憾的超时提交了。错误点主要在于,这里求的是第K大的数,而若是我们使用K去判断快排得到的下标,得到的是第K个数(等同于排序以后从左往右下标为K-1),而题中隐藏的意思等同于排序以后从 右往左数第K个数。所写在写代码的时候要注意一个K值得转换。
#include<iostream>
#include<vector>
#include<algorithm> using namespace std; int patiton(vector<int> &ivec,int left,int right);
int main()
{
vector<int> ivec;
int x;
while(cin>>x)
{
ivec.push_back(x);
}
int K;
cin>>K; int left=,right=ivec.size()-;
int index=patiton(ivec,left,right);
K=ivec.size()-K; //注意这行
while(index !=K)
{
if(index<K)
left=index+;
else
right=index-;
index=patiton(ivec,left,right);
}
cout<<ivec[K];
return ;
} int patiton(vector<int> &ivec,int left,int right)
{
int i=left-; int pivot=ivec[right];
for(int j=left;j<right;j++)
{
if(ivec[j]<=pivot)
{
i++;
swap(ivec[i],ivec[j]);
}
}
swap(ivec[i+],ivec[right]);
return i+;
}
注意:在本地IDE编译的时候,while(cin>>x)会存在一个退出的问题。这个可以上网百度一下就行。