链接:https://ac.nowcoder.com/acm/contest/5773/A
来源:牛客网
给你一个长度为n的序列,求序列中第k小数的多少。
输入描述:
多组输入,第一行读入一个整数T表示有T组数据。
每组数据占两行,第一行为两个整数n,k,表示数列长度和k。
第二行为n个用空格隔开的整数。
输出描述:
对于每组数据,输出它的第k小数是多少。
每组数据之间用空格隔开
示例1
输出
复制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]); } }