其他分享
首页 > 其他分享> > 786. 第k个数

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<iostream>
#include<cmath>

using namespace std;

const int N=1e6+10;

int n,k;
int a[N];

void quick_sort(int a[],int l,int r){
    if(l>=r) return;
    
    int i=l-1,j=r+1,x=a[(l+r)>>1];
    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,l,j);
    quick_sort(a,j+1,r);
}

int main(){
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    
    quick_sort(a,1,n);
    
    printf("%d\n",a[k]);
    
    return 0;
}

  

标签:输出,786,数列,int,样例,个数,整数,include
来源: https://www.cnblogs.com/Iamcookieandyou/p/13340503.html