求第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;
}

 

posted on 2014-09-17 09:58  后端bug开发工程师  阅读(2284)  评论(0编辑  收藏  举报

导航