快排的思想求第K小的数

描述

输入N个数，M次查询。

每次查询给出一个数x。

 

要求：每次查询输出前x个数中第i小的数。（i为第i次查询）

你可以假设M  <= N，Xi <= Xi+1 <= Xi+2 <= ……. <= Xm (Xm <= N).

输入

Line0:T
Line1: N,M
Line2…LineN+1：num1,......,numN
LineN+2…LineN+2+M:x1,……,xM

N < 30000, num < 2000000000

输出

每次查询输出前i小的数，单独一行。
详细格式请参考样例。

样例输入

1
7 4
3 1 -4 2 8 -1000 2
1 2 6 6

样例输出

3
3
1
2

可以根据快排的思想确定第i小，思考快排的过程，每次进行一次划分，数列左边的数都小于标杆值，数列右边的值都大于标杆值，这就已经可以确定标杆值是第r小，加入求第k小，如果r<k，则在左半部分找，r=k，则直接返回，r>k，则在右半部分找，每次去掉一半，这是减治法的思想，复杂度为log2 n

#include <iostream>
#include <cstdio>
using namespace std;

int mypartition(int arr[],int start,int dest);
int quickSort(int arr[],int start,int dest,int k);

int main()
{
    #ifdef LOCAL_DEBUG
    freopen("input.txt","r",stdin);
    #endif // LOCAL_DEBUG
    
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n,m;
        scanf("%d%d",&n,&m);
        for(int i=0;i<n;i++) {
            scanf("%d",&a[i]);
        }
        int x;
        for(int i=0;i<m;i++)
        {
            scanf("%d",&x);
            int ans = quickSort(a,0,x-1,i);
            printf("%d\n",ans);
        }
    }
    return 0;
}

int quickSort(int arr[],int start,int dest,int k)
{
    int middle=mypartition(arr,start,dest);
    if(middle==k) return arr[k];
    if(middle>k)
        return quickSort(arr,start,middle-1,k);
    if(middle<k)
        return quickSort(arr,middle+1,dest,k);
}

int mypartition(int arr[],int start,int dest)
{
    int i=start;
    int j=dest;
    while(i<j)
    {
        while(i<j && arr[i]<=arr[j])
            j--;
        if(i<j)
        {
            std::swap(arr[i],arr[j]);
            i++;
        }
        while(i<j && arr[i]<=arr[j])
            i++;
        if(i<j)
        {
            std::swap(arr[i],arr[j]);
            j--;
        }
    }
    return j;
}