第k个数和快速排序代码模板差不多,只是在函数调用上加一个参数k,并在递归时判断一下。
思路:
- 先进行一次快排,使x左侧都<=x,x右侧都>=x;
- 计算x左侧数的个数:ls=j-l+1(注意+1)
- 判断k在x哪侧
- ls>=k,则k在x左侧,只需递归x左侧
- ls<k,则k在x右侧,只需递归x右侧
题目:
代码:
#include<iostream>
using namespace std;
const int N=1e5+10;
int n,k;
int q[N];
int quick_sort(int l,int r,int k){
if(l>=r)return q[l];
int x=q[(l+r)>>1],i=l-1,j=r+1;
while(i<j){
while(q[++i]<x);
while(q[--j]>x);
if(i<j)swap(q[i],q[j]);
}
int ls=j-l+1; //判断x左侧个数
if(ls>=k)return quick_sort(l,j,k); //k在x左侧,只需递归x左侧
return quick_sort(j+1,r,k-ls); //k在x右侧,只需递归x右侧
}
int main(){
cin>>n;
cin>>k;
for(int i=0;i<n;i++)cin>>q[i];
cout<<quick_sort(0,n-1,k);
return 0;
}
有关算法竞赛的题目会继续更新,欢迎评论交流!