第k小数 快排变种

题目链接

手写快排，每次排序判断基准m,的位置，如果m > k ，第k小数就在
[l ,m-1],如果m < k,第k小数就在[m + 1,r] ,否则m == k,num[m]就是第k小数。

数据量偏大需要自己写个快读，不然可能读入数据过程就超限了

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll num[6000000];
ll T,n,k;

void read(ll &x){
    char c;
    ll flag = 1;
    x = 0;
    while((c = getchar()) && (c < '0' || c > '9'))if(c =='-')flag = -1;
    x = c - '0';
    while((c = getchar()) && (c >= '0' && c <= '9')) x = x*10 + c -'0';
    x*=flag;
}

ll qsort(ll l,ll r){
    
    if(l == r)   return num[l];
    ll m = l,head = l,rear = r;
    num[0] = num[m];
    
    while(head < rear){
        while(num[0] <= num[rear] && rear > m)    rear--;
             num[m] = num[rear];// '>='两个数一样不用交换 rear > m 防止越界
             m = rear;
        
        while(num[0] >= num[head] && head < m)    head++;
            num[m] = num[head];
            m = head;
    }
    
    num[m] = num[0];
   
    if(m < k)
        return qsort(m + 1,r);
    
    else if(m > k)
        return qsort(l,m-1);
    
    return num[k];
    
}

int main(){
   
    read(T);
    for(int t = 0;t < T;t++){
        read(n); read(k);
        for(int i = 1;i <= n;i++)
            read(num[i]);
        cout << qsort(1,n) << endl;
    }
    return 0;
}