输出前K大的数

关于问题

描述

给定一个数组,统计前k大的数并且把这k个数从大到小输出。

输入

第一行包含一个整数n,表示数组的大小。n < 100000。
第二行包含n个整数,表示数组的元素,整数之间以一个空格分开。每个整数的绝对值不超过100000000。
第三行包含一个整数k。k < n。

输出

从大到小输出前k大的数,每个数一行。

样例输入

10
4 5 6 9 8 7 1 2 3 0
5

样例输出

9
8
7
6
5

解决方法

排序(应用快速排序)后输出,复杂度(nlogn)

分治处理,复杂度(n+klogk)

思路:把前k大的都弄到数组最右边,然后对最右边这k个进行排序输出

关键:在O(n)时间内把前k个都弄到数组最右边

具体实现:

(设数组起始下标为start,结束下边为end,基准元素下标为privot)
1.应用快速排序的思想,一趟快速排序后,privot左边的所有元素都<=privot,privot右边的所有元素都>=privot,
2.此时若privot右边的元素个数(end-privot+1)刚好等于K就rerturn
如果右边的元素个数少于K,那么在[start,privot-1]的范围内再找前(k-(end-privot+1))大的数
否则就继续在[privot+1,end]的范围内找前k大的数

时间复杂度分析

每次都是选择数组的前部分或者后一部分进行操作,我们在这里假设每次都恰好是一半
T(n) = T(n/2) + a*n
= T(n/4) + a*n/2 + a*n
= T(n/8) + a*n/4 + a*n/2 + a*n
= … …
= T(1) + … …+ a*n/8 + a*n/4 + a*n/2 + a*n
< 2*a*n
即 O(n)

代码实现

#include<iostream>
#include<cstdio>
using namespace std;
int k;
int a[100005];
void swap(int & a,int & b)
{
    int tmp;
    tmp = a;
    a = b;
    b = tmp;
}

void Qsort( int a[], int s, int e )
{ 
    if(s >= e)  
        return;
    int k = a[s];  
    int i = s,j = e;
    while (i < j) 
    { 
        while ( j > i && a[j] >= k )
            --j;
        swap( a[i], a[j]);
        while ( i < j && a[i] <= k )
            ++i;
        swap( a[i], a[j]);

    }
    Qsort( a, s, i-1 );     
    Qsort( a, i+1, e );    
}

void arrangeRight(int a[],int k,int s,int e)
{
    if(s <= e)
    {
        int kk = a[s];  
        int i = s,j = e;
        while (i < j) 
        { 
            while ( j > i && a[j] >= kk )
                --j;
            swap( a[i], a[j]);
            while ( i < j && a[i] <= kk )
                ++i;
            swap( a[i], a[j]);

        }
        if(e - i + 1 == k)  return;
        else if(e - i + 1 > k)  arrangeRight(a,k,i+1,e);
        else                    arrangeRight(a,k-(e-i+1),s,i-1);
    }

} 

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

注:部分思想引用于郭炜老师的《程序设计与算法二》

发布了48 篇原创文章 · 获赞 17 · 访问量 2万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 编程工作室 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览