设当前划分的下标为ind.
如果ind+1==k,直接返回a[ind]
如果ind+1<k,递归进入[ind+1,r)的区间继续寻找答案
接下来就是处理重复元素的关键步骤,如果ind+1>k
可分成两种情况:
1、k位于重复元素[ind+1-cnt+1,ind+1]之中,直接返回a[ind],直接结束程序.
2、k位于所有重复元素之前,则应该丢弃重复元素,递归进入[l,ind-cnt+1)的区间继续寻找答案
当然,这题n<=10^6,直接用sort以O(nlgn)也能过。
AC代码:
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=1e6+5;
int a[maxn];
int n,k;
inline int findmid(int l,int r){ //中位数的中位数
if(r-l<=5) return (l+r)/2;
for(int i=0;i<(r-l)/5;++i){
sort(a+l+i*5,a+l+i*5+5);
swap(a[l+i],a[l+i*5+2]);
}
return findmid(l,l+(r-l)/5);
}
int partion(int l,int r,int &p){ //改进版partion
int h=findmid(l,r);
swap(a[h],a[r-1]);
p=0;
int ind=l-1;
for(int i=l;i<r-1;++i){
if(a[i]==a[r-1]) ++p;
if(a[i]<=a[r-1])
swap(a[++ind],a[i]);
}
++p;
swap(a[++ind],a[r-1]);
int i=l,j=ind-1;
while(i<j){
if(a[i]==a[ind]){
while(a[j]==a[ind]) --j;
if(i<j){
swap(a[i],a[j]);
--j;
}
}
++i;
}
return ind;
}
int solve(int l,int r){
int p=0;
int ind=partion(l,r,p);
if(ind+1==k) return a[ind];
if(ind+1>k){
if(ind+1-p+1<=k) return a[ind];
else return solve(l,ind-p+1);
}
if(ind+1<k) return solve(ind+1,r);
}
int main(){
scanf("%d%d",&n,&k);
for(int i=0;i<n;++i) scanf("%d",&a[i]);
printf("%d\n",solve(0,n));
return 0;
}
如有不当之处欢迎指出!