K番目の10進数のクイックソートバリアント

トピックリンク

手書きのクイックソート。ソートがベンチマーク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;
}