题目描述
给定一个长度为 n 的整数数列,以及一个整数 k,请用快速选择算法求出数列从小到大排序后的第 k个数。
输入格式
第一行包含两个整数 n 和 k。
第二行包含 n个整数(所有整数均在 1∼10^9范围内),表示整数数列。
输出格式
输出一个整数,表示数列的第 k� 小数。
数据范围
1≤n≤100000
1≤k≤n
输入样例:
5 3
2 4 1 5 3
输出样例:
3
思路:首先排序用快排,但是快排的核心思想是,x前面的数,全都小于等于x,后面的数全都大于等于x,然后递归,我们只用看x前面的数是大于k还是小于k,如果大于k,那么只用递归x前面的数即可,如果小于k,那么就需要递归x后面的数,不过此时应该是第k-sl个数了。
故步骤为:
1.确定划分x,
2.循环使x前面的数小于等于x,后面的大于等于x
3.确定递归哪部分
代码:
#include<bits/stdc++.h>
using namespace std;
int a[100010];
int quick_sort(int l,int r,int k)
{
if(l==r) return l;//只有一个数
int i=l-1,j=r+1,x=a[(l+r)/2];
while(i<j)
{
while(a[++i]<x);
while(a[--j]>x);
if(i<j) swap(a[i],a[j]);
}
int sl=j-l+1;
if(sl<k) return quick_sort(j+1,r,k-sl);//递归右半边
else return quick_sort(l,j,k);//递归左半边
}
int main()
{
int n,k;
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
int m=quick_sort(1,n,k);
printf("%d",a[m]);
}