第一种递归的时候思路是找第k个数,然后边界处的return是当时递归找到的那一个数(就是第k个数)。所以需要定义一个sl,左右递归的K也会不同
C++ 代码
#include <iostream>
using namespace std;
int q[100010];
int Qsort(int l,int r,int k){
if(l==r) return q[l];
int i=l-1,j=r+1,x=q[l+r>>1];
while(i<j){
do i++;while(q[i]<x);
do j--;while(q[j]>x);
if(i<j) swap(q[i],q[j]);
}
int sl=j-l+1;
if(k<=sl) Qsort(l,j,k);
else Qsort(j+1,r,k-sl);
}
int main(){
int n,k;
cin>>n>>k;
for(int i=0;i<n;i++) cin>>q[i];
cout<<Qsort(0,n-1,k);
return 0;
}
第二种递归的时候思路是找下标为K的数,边界处的return是q[k].所以不需要定义sl,左右递归的K也相同。
C++ 代码
#include <iostream>
using namespace std;
int q[100010];
int Qsort(int l,int r,int k){
if(l==r) return q[k];
int i=l-1,j=r+1,x=q[l+r>>1];
while(i<j){
do i++;while(q[i]<x);
do j--;while(q[j]>x);
if(i<j) swap(q[i],q[j]);
}
if(k<=j) Qsort(l,j,k);
else Qsort(j+1,r,k);
}
int main(){
int n,k;
cin>>n>>k;
for(int i=0;i<n;i++) cin>>q[i];
cout<<Qsort(0,n-1,k-1);
return 0;
}
总的来说第二种更方便