第K小的数(快排)

#include <iostream>
using namespace std;
int n,k;
int a[10001];
int findk(int k,int* arr,int l,int r) {
    if (l == r) return a[l];//或者a[r]; 
    int i=l,
        j=r;
    int mid=(l+r)/2;
    while (i<=j) {
        while (arr[i] < arr[mid]) i++;//如果加了=就会无限循环 
        while (arr[j] > arr[mid]) j--;//假设arr[mid]是最大的值 那么还会继续找(找不到) 
        if (i <= j) {
            swap(arr[i],arr[j]);
            i++;j--;
        }
    }
    if (i <= k) findk(k,arr,i,r);
    else if (j >= k) findk(k,arr,l,j);
    else return arr[mid];
}
int main() {
    cin>>n;
    for (int i=1;i<=n;i++)
        cin>>a[i];
    cin>>k;
    cout<<findk(k,a,1,n)<<endl;
    return 0;
}