（第k大）第k小问题

链接：https://ac.nowcoder.com/acm/contest/5773/A
来源：牛客网

给你一个长度为n的序列，求序列中第k小数的多少。

输入描述:

多组输入，第一行读入一个整数T表示有T组数据。

每组数据占两行，第一行为两个整数n，k，表示数列长度和k。

第二行为n个用空格隔开的整数。

输出描述:

对于每组数据，输出它的第k小数是多少。

每组数据之间用空格隔开

示例1

输入

复制 2 5 2 1 4 2 3 4 3 3 3 2 1

2
5 2
1 4 2 3 4
3 3
3 2 1

输出

复制 2 3

2
3

对于这种问题，快排和sort被卡了，正解应该为堆排序（因为在排序过程中便可以输出），其实还有一直方式就是
用stl中自带的nth_element(a,a+k-1,a+n);
这样可以求第k小的数字，如果要输出第k大的则可以nth_(a,a+n-k,a+n)

#include<bits/stdc++.h>
using namespace std;
int n,k,t;
int a[5000005];
inline int read(){
    int x = 0, f = 1;
    char ch = getchar();
    while(ch < '0' || ch > '9'){
        if (ch == '-')
            f = -1;
        ch = getchar();
    }
    while(ch >= '0' && ch <= '9'){
        x = (x<<1) + (x<<3) + (ch^48);
        ch = getchar();
    }
    return x * f;
}
int main()
{
    t=read();
    while(t--)
    {
        n=read();
        k=read();
        for(int i=0;i<n;i++)
        {
            a[i]=read();
        }
        nth_element(a,a+k-1,a+n);//STL
        printf("%d\n",a[k-1]);
    }

    
}