第K小的数

122 阅读1分钟

 

输入 nnn(1≤n<50000001 \le n < 50000001≤n<5000000 且 nnn 为奇数)个数字 aia_iai​(1≤ai<1091 \le a_i < {10}^91≤ai​<109),输出这些数字的第 kkk 小的数。最小的数是第 000 小。

输入格式

输出格式

输入输出样例

输入 #1

5 1
4 3 2 1 5

输出 #1

2

个人思路:

这里介绍一个好用的函数,C++ nth_element(),看名字我们也能感觉出来, 这个函数能帮我们找到第k小的数,当然,如果我们自己写一个降序排序规则,也可以求第k大的数。

/排序规则采用默认的升序排序
void nth_element (RandomAccessIterator first,
                  RandomAccessIterator nth,
                  RandomAccessIterator last);
//排序规则为自定义的 comp 排序规则
void nth_element (RandomAccessIterator first,
                  RandomAccessIterator nth,
                  RandomAccessIterator last,
                  Compare comp);

  • first 和 last:都是随机访问迭代器,[first, last) 用于指定该函数的作用范围;
  • nth:也是随机访问迭代器,其功能是令函数查找“第 nth 大”的元素,并将其移动到 nth 指向的位置;
  • comp:用于自定义排序规则。当然可以省略。

AC代码:

#include<bits/stdc++.h>
using namespace std;
int x[5000005],k;
int main()
{
	int n;
	scanf("%d%d",&n,&k);
	for(int i=0;i<n;i++)
		scanf("%d",&x[i]);
	nth_element(x,x+k,x+n);//简短又高效
	printf("%d",x[k]);
}