题目描述
给定一个长度为n的整数数列,以及一个整数k,请用快速选择算法求出数列从小到大排序后的第k个数。
输入格式
第一行包含两个整数 n 和 k。
第二行包含 n 个整数(所有整数均在1~109范围内),表示整数数列。
输出格式
输出一个整数,表示数列的第k小数。
数据范围
1≤n≤100000,
1≤k≤n
输入样例
5 3
2 4 1 5 3
输出样例
3
算法1(快速排序)
C++ 代码
#include<iostream>
#include<algorithm>
using namespace std;
const int N=1e5+10;
int n,k;
int a[N];
int kp(int l,int r,int k){
if(l==r){
return a[l];
}
if(l<r){
int i=l-1;
int j=r+1;
int x=a[l];
while(i<j){
while(a[--j]>x);
while(a[++i]<x);
if(i>=j){
break;
}
swap(a[i],a[j]);
}
int cnt=j-l+1;
if(k<=cnt){
return kp(l,i,k);
}
return kp(j+1,r,k-cnt);
}
}
int main()
{
cin >> n >> k;
for(int i=1;i<=n;i++){
cin >> a[i];
}
cout << kp(1,n,k) << endl;
return 0;
}
算法2(nth_element())
C++ 代码
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
const int N=1e5+5;
int n,k;
int a[N];
int main()
{
cin >> n >> k;
for(int i=1;i<=n;i++){
cin>> a[i];
}
nth_element(a,a+k,a+n+1);
cout << a[k] << endl;
return 0;
}
算法3(STL sort)
C++ 代码
#include<iostream>
#include<algorithm>
using namespace std;
const int N=1e5+5;
int n,k;
int a[N];
int main()
{
cin >> n >> k;
for(int i=1;i<=n;i++){
cin >> a[i];
}
sort(a+1,a+1+n);
cout << a[k] << endl;
return 0;
}