【大旭刷题笔记】之【第k小的数】

寻找第k小的数模板

2022-1025
记录一下自己的刷题历程！精进不休，Fighting！
链接: Acwing题目链接
今天的题目是第k小的数，是基于昨天的快速排序的改进算法，快速找到第k小的数，第k大的数同理，只要更改一下判断条件即可。
代码模板如下：

#include<iostream>
using namespace std;
const int N=100010;
int d[N];
int k;

int quickSort(int d[], int l, int r,int k){
    if(l>=r) return d[l];
    
    int i=l-1,j=r+1,x=d[(l+r)>>1];
    while(i<j){
        do i++;while(d[i]<x);
        do j--;while(d[j]>x);
        if(i<j) swap(d[i],d[j]);
    }
    //若i=j时左侧的元素个数大于等于k，就不用排i=j右侧的元素了
    if (j-l+1>=k) return quickSort(d,l,j,k);
    //若i=j时左侧的元素个数小于k个，说明第K小的数一定在i=j位置的右侧，此时只用
    //排j+1到最后的元素即可，此时的k更新为k-（j-l+1）j-l+1表示j右侧的元素个数（包含j）
    else return quickSort(d,j+1,r,k-(j-l+1));
}

int main(){
    int n;
    cin>>n>>k;
    for(int i=0;i<n;++i){
        cin>>d[i];
    }
   
    cout<<quickSort(d,0,n-1,k)<<endl;
    return 0;
}在这里插入代码片