AcWing 786. 第k个数

⬅ 返回

1.题目链接

给定一个长度为 n 的整数数列，以及一个整数 k，请用快速选择算法求出数列从小到大排序后的第 k 个数。

输入格式

第一行包含两个整数 n 和 k。

第二行包含 n 个整数（所有整数均在 1∼109 范围内），表示整数数列。

输出格式

输出一个整数，表示数列的第 k 小数。

数据范围

1≤n≤100000,

1≤k≤n

输入样例：

5 3
2 4 1 5 3

输出样例：

3

2.代码

#include <iostream> 
#include <algorithm>
 
using namespace std;
const int N = 1e6+120;
 
int a[N];
 
void quick_sort(int a[],int l,int r){
	if(l>=r)return;
	int x = a[(l+r)/2];
	int i=l-1,j=r+1;
	while(i<j){
		do i++;while(a[i]<x);
		do j--;while(a[j]>x);
		if(i<j)swap(a[i],a[j]);
	}
	quick_sort(a,l,j);
	quick_sort(a,j+1,r);
	
}
 
 
 
int main(){
	int n,k;
	cin>>n>>k;
	for(int i=0;i<n;i++)
	{
		cin>>a[i];
	}
	quick_sort(a,0,n-1); 
	cout<<a[k-1]<<endl;
	
}