首页 > 代码库 > 求第k大数

求第k大数

题意:输入n个整数和一个正整数k(k>=1&&k<=n),输出这些数从小到大排序后的第k个(例如,k=1就是最小值)。n<=10^7。

分析:由于n比较大如果先排序后求解,那时间复杂度有点高了n*log(n),然后hash的话数组开不了,用什么方法求解呢?我这里给出一种用快速排序的方法解决,当然这就需要对快速排序的原理理解了,自己百度吧!

代码实现:

#include<iostream>#include<cstdio>#include<cstring>#include<cmath>#include<algorithm>using namespace std;int k;int a[10000005];int partition(int l,int r,int a[]){    int key=a[l],index=l;    for(int i=l+1;i<r;i++)     if(a[i]<key)     {         index++;         swap(a[index],a[i]);     }     swap(a[l],a[index]);    return index;}void quickSort(int l, int r, int a[]){    if(l<r)    {        int mid = partition(l,r,a);        if(k>=l&&k<=mid-1)         quickSort(l,mid-1,a);        else         quickSort(mid+1,r,a);    }}int main(){    int i,n;    while(scanf("%d%d",&n,&k)!=EOF)    {        k--;        for(i=0; i<n; i++)         scanf("%d",&a[i]);        quickSort(0,n,a);        for(i=0;i<n-1;i++)         printf("%d ",a[i]);        printf("%d\n",a[n-1]);        printf("%d\n",a[k]);    }    return 0;}

 

求第k大数