数据结构：寻找第k小的数

七月 ღ

已于 2022-02-27 21:00:03 修改

阅读量2.1k

点赞数

文章标签： c语言 c++ 算法

于 2022-02-27 20:55:57 首次发布

本文链接：https://blog.csdn.net/weixin_46101356/article/details/123170178

版权

这篇博客探讨了如何在大量整数中快速找到第k小的数，提出了利用排序和STL中的nth_element函数来解决这一问题。输入数据可能非常大，因此输入处理被优化以减少耗时。代码示例展示了如何读取和处理数据，并输出第k小的数。

摘要由CSDN通过智能技术生成

寻找第k小的数

题目

给定若干整数，请设计一个高效的算法，确定第k小的数。
输入格式:
测试数据有多组，处理到文件尾。每组测试数据的第1行输入2个整数n，k（1≤k≤n≤1000000）。
第2行输入n个整数，每个数据的取值范围在0到1000000之间。
输出格式:
对于每组测试，输出第k小的数。
输入样例:

5 3
1 2 2 2 1
9 3
1 2 3 4 5 6 9 8 7

输出样例:

2
3

提示:
需要设计的是高效的算法！
可以使用STL之sort、nth_element函数求解。
另外，由于输入数据特别多，为减少输入数据的耗时，建议使用以下的输入外挂函数：

void scan_d(int &num)
{
char in;
in=getchar();
while(in<‘0’||in>‘9’) in=getchar();
num=in-‘0’;
while(in=getchar(),in>=‘0’&&in<=‘9’){num*=10,num+=in-‘0’;}
}
调用方法如下：

int n;
scan_d(n);

代码

#include<iostream>
#include<algorithm>
using namespace std;
int a[1000001];
int main(){
	int n,k;
	while(cin>>n>>k){
		for(int i=1;i<=n;i++){
			scanf("%d",a+i);
		}
		sort(a+1,a+n+1);
		//nth_element（a+1,a+k,a+n+1);
		cout<<a[k]<<endl;
	}
}