输出前k大的数

010:输出前k大的数

总Time Limit: 10000ms 单个测试点时间限制: 1000ms Memory Limit: 65536kB

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

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

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

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

Sample Output
9
8
7
6
5

解题思路：

 

排序后再输出，复杂度 O(nlogn)
用分治处理：复杂度 O(n+mlogm)

思路：把前m大的都弄到数组最右边，然后对这最右边m个元素排序， 再输出

关键 ：O(n)时间内实现把前m大的都弄到数组最右边

引入操作 arrangeRight(k): 把数组(或数组的一部分）前k大的 都弄到最右边
                                        
如何将前k大的都弄到最右边 
    1）设key=a[0], 将key挪到适当位置，使得比key小的元素都在 key左边，比key大的元素都在key右边（线性时间完成）
    2)  选择数组的前部或后部再进行 arrangeRight操作 

        
        a = k ————done 
        a > k ————对最右边a-1个元素再进行arrangeRigth(k) 
        a < k ————对左边b个元素再进行arrangeRight(k-a)

将前m大的都弄到数组最右边的时间：
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>
#include <algorithm>

using namespace std;

int a[100010];

void arrangeRight(int s,int e,int k) //把数组的前 k 大的元素都弄到最右边
{
    if( s - e +1 == k ) //也别忽略了这一步
		return;

    int key = a[s]; //就从第一个元素开始，以第一个元素为判断参照

    int i = s ,j = e; //确定好左右指针
    while( i != j ) //大条件
    {
        while( j > i &&  a[j] >= key ) //后面的元素比 k 大：符合情况
            --j;
        swap(a[i],a[j]); //正是因为找到了后面的元素比 K 小了，所以要进行置换
        while( i < j && a[i] <= key ) //前面的元素比 k 小：符合情况
            ++i;
        swap(a[i],a[j]); //正是因为找到了前面的元素比 k 大了，所以要进行置换
    } //处理完后，a[i] = key

    int sum = e+1 - i;//数组中从key开始往右的元素个数和
    if(sum == k) //完成
        return ;
    else if(sum > k) //对最右边的 sum-1 个元素再进行arrangeRight(k)
        arrangeRight(i+1,e,k);
    else if(sum < k)
        arrangeRight(s,i-1,k-sum); // 对左边 b(key之前的元素个数) 个元素再进行arrangeRight(k-sum)

}

int main()
{
    int n,k;
    scanf("%d",&n);
    for(int i = 0; i<n ;i++)
    {
        scanf("%d",&a[i]);
    }
    scanf("%d",&k);
    arrangeRight(0,n-1,k); //把数组的前 k 大的元素都弄到最右边
    sort(a+n-k,a+n);
    for(int i = n-1; i >= n-k ; --i) //输出
        cout << a[i] << endl;
    return 0;
}