第k小数 (C++ STL)

#include<bits/stdc++.h>
using namespace std;
const int N = 5e6+10;
int T,n,k,a[N];
int main()
{
    ios::sync_with_stdio(false);
    cin>>T;
    while(T--)
    {
        cin>>n>>k;
        for(int i=0;i<n;i++) cin>>a[i];
        nth_element(a,a+k-1,a+n);
        cout<<a[k-1]<<endl; 
    }
    return 0;
}