第k小数--牛客

题目：
给你一个长度为n的序列，求序列中第k小数的多少。
输入描述:
链接：第k小数–牛客
来源：牛客网

多组输入，第一行读入一个整数T表示有T组数据。每组数据占两行，第一行为两个整数n，k，表示数列长度和k。第二行为n个用空格隔开的整数。
输出描述:
对于每组数据，输出它的第k小数是多少。每组数据之间用空格隔开。
示例1
输入：
2
5 2
1 4 2 3 4
3 3
3 2 1
输出：
2
3

思路： 这道题可能会有大量的输入数据，普通的输入可能会超时，这道题需要用到快读和nth_element()函数。
快排：

#define re register
inline int read()
{
    re int x=0,f=1;
    re char ch=getchar();
    while(ch<'0'||ch>'9') {if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
    return x*f;
}

nth_element函数
用法：
nth_element(start,start+k,end)
使得第k大的数安排在下标为k的数组上。

代码如下：

#include <iostream>
#include <algorithm>
using namespace std;
#define re register
inline int read()
{
    re int x=0,f=1;
    re char ch=getchar();
    while(ch<'0'||ch>'9') {if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
    return x*f;
}
int main()
{
 const int maxn=5e6+5;
 int i,t,n,k,a[maxn];
 cin>>t;
 while(t--)
 {
  n=read();
        k=read();
  for(i=1;i<=n;i++)
  {
   a[i]=read();
  }
  nth_element(a,a+k,a+n+1);
  cout<<a[k+1]<<endl;
 }
 return 0;
}