寻找第k小的数字

这个代码要求输入的数组不存在重复值

在快速排序的过程中，我们就可以实现寻找第k小的数组

时间复杂度O(logn)

#include <bits/stdc++.h>
using namespace std;
#define int long long 
#define vi vector<int>
#define sz(x) (int)x.size()
const int N=101010;
//2023.3.15
vi a(N);
int finds(int l,int r,int k){
    if(l>=r)return a[l];
    int i=l+1,j=r,val=a[l];//将a[l]设置为基准值
    while(true){//对于l-r区间进行快速排序
        while(a[i]<val)i++;
        while(a[j]>val)j--;
        if(i>=j)break;
        swap(a[i],a[j]);
    }
    if(j-l+1==k)return val;//如果刚好这个基准值就是第k小的数，直接返回，运气太好了
    a[l]=a[j];
    a[j]=val;
    if(j-l+1<k)return finds(j+1,r,k-(j-l+1));//如果小于k，就要在右边寻找，第k-（j-l+1）小的值
    else return finds(l,j-1,k);//否则在左面寻找第k小的数
}

void solve(){
    int n,k;cin>>n>>k;
    for(int i=0;i<n;i++)cin>>a[i];
    cout<<finds(0,n-1,k);
}

signed main()
{
    ios::sync_with_stdio(false),cin.tie(nullptr),cout.tie(nullptr);
    solve();
    return 0;
}