题目: AcWing 786. 第k个数
给定一个长度为 n 的整数数列,以及一个整数 k,请用快速选择算法求出数列从小到大排序后的第 k 个数。
输入格式:
第一行包含两个整数 n 和 k。
第二行包含 n 个整数(所有整数均在 1∼109 范围内),表示整数数列。
输出格式:
输出一个整数,表示数列的第 k 小数。
数据范围:
1≤n≤100000,
1≤k≤n
输入样例:
5 3
2 4 1 5 3
输出样例:
3
题目分析:
简单做法就是使用排序算法直接排序,如何输出第k个数即可。
下面我们从快速排序找到灵感:
快速排序的步骤就是
第一步:找到基准值
第二步:将数组分成两部分(小于等于基准值,大于等于基准值)
第三步:递归两部分(直到是一个数字)
如果我们知道第k个值在哪一部分里,我们是不是就会少排序一部分,这样层层递归下去直到是一个数(那么这个数就是第k的数),这个方法就像二分查找一样一点一点减少数据量(大大减少时间复杂度)
#include <iostream>
#include <algorithm>
using namespace std;
const int N=100000+5;
int p[N];
int quick_sort(int p[],int l,int r,int k)
{
if(l==r)return p[l];
int i=l-1,j=r+1,x=p[l+r>>1];
while(i<j)
{
do i++;while(p[i]<x);
do j--;while(p[j]>x);
if(i<j)swap(p[i],p[j]);
}
if(k<=j)
{
return quick_sort(p,l,j,k);
}else{
return quick_sort(p,j+1,r,k);
}
}
int main()
{
int n,k;
scanf("%d%d",&n,&k);
for(int i = 0; i < n;i ++)scanf("%d",&p[i]);
cout<< quick_sort(p,0,n-1,k-1)<<endl;
return 0;
}