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
程序:
#include<cstdio>
#include<algorithm>
using namespace std;
int a[1000005],i,j,n,x,m;
void quick_sort(int a[],int down,int up)
{
if(down>=up)return;
i=down-1,j=up+1,x=a[(down+up)/2];
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,down,j);
quick_sort(a,j+1,up);
}
int main()
{
scanf("%d%d",&n,&m);
for(i=0;i<n;i++)scanf("%d",&a[i]);
quick_sort(a,0,n-1);
printf("%d\n",a[m-1]);
return 0;
}