第K个数

给定一个长度为n的整数数列,以及一个整数k,请用快速选择算法求出数列的第k小的数是多少。

输入格式

第一行包含两个整数 n 和 k。

第二行包含 n 个整数(所有整数均在1~109

范围内),表示整数数列。

输出格式

输出一个整数,表示数列的第k小数。

数据范围

1≤n≤100000
1≤kn

输入样例:
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]);
}
Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐