第K个数

第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<stdio.h>
#include<iostream>
using namespace std;
const int N = 100001;
void quick_sort(int p[],int l,int r)
{
    if(l>=r) return;
    int i=l-1, j=r+1,x=p[(l+r)/2];
    while(i<j)
    {
        do i++; while(p[i]<x);
        do j-- ;while(p[j]>x);
        if(i<j) swap(p[i],p[j]);
    }
    quick_sort(p,l,j);
    quick_sort(p,j+1,r);
}
int main()
{
    int n,m;
    int p[N];
    scanf("%d %d",&n,&m);
    for(int i=0;i<n;i++) scanf("%d",&p[i]);
    quick_sort(p,0,n-1);
    printf("%d",p[m-1]);
}