O(n)线性时间找第K大,中位数

运用快速排序的思想,可以达到线性时间找到一串数的第K大

 #include<cstdio>
#define F(i,a,b) for(int i=a;i<=b;i++) int a[],n;
void swap(int &a,int &b){int c=a;a=b,b=c;} int partition(int *a,int l,int r){
int x=a[l],pos=l;
for(int i=l+;i<=r;i++)
if(a[i]<x)swap(a[++pos],a[i]);
swap(a[l],a[pos]);
return pos;
} int find_k(int *a,int k,int l=,int r=n){
int pos=partition(a,l,r);
int j=pos-l+;
if(l==r)return a[l];
if(k==j)return a[pos];
if(k>j)return find_k(a,k-j,pos+,r);
else return find_k(a,k,l,pos-);
} double findmid(){
if(n&){
int k=(n>>)+;
double ans=find_k(a,k);
return ans;
}else{
int k=n>>;
double ans=find_k(a,k)+find_k(a,k+);
return ans/;
}
} int main(){
scanf("%d",&n);
F(i,,n)scanf("%d",a+i);
printf("mid=%lf\n",findmid());
return ;
}
上一篇:关于JQ中ready()方法的几种写法总结


下一篇:C语言回顾-字符串指针