输出前k大的数

题目描述

给定一个数组包含n个元素，统计前k大的数并且把这k个数从大到小输出。

输入

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

输出

从大到小输出前k大的数。

样例输入

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

样例输出

10 9 8 7 6

思路1：排序后再输出。复杂度O(nlogn)
具体细节没什么好说的，直接调用algorithm中的sort();

思路2：分治。复杂度O(n+mlogm)
把前k大的数放到数组右边，然后对右边这k个数sort();
那如何在O(n)的时间内把前k大的数放到数组右边呢？想想快速排序吧 : )

代码

//输出前k大的数 
#include <iostream>
#include <algorithm>
using namespace std;

void swap(int &a, int &b){
    int tmp = a;
    a = b;
    b = tmp;
}

//在数组a中找前k大的数 
void arrangeRight(int a[], int s, int e, int k){
    if(s-e+1 == k){
        return ;
    }
    //以第一个数为基准，大于base的放在数组右边，小的放左边 
    int base = a[s];
    int i = s;
    int j = e;
    while(i != j){
        while(i < j && a[j] >= base){
            j--;
        }
        while(i < j && a[i] <= base){
            i++;
        }
        if(i < j){
            swap(a[i], a[j]);
        }
    }
    //最后i==j，将其与第一个基准点交换
    swap(a[i], a[s]);

    //i右边个数 
    int right = e-i+1;

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

int main(){
    int a[10];
    int n; //共n个数 
    cin>>n;
    for(int i=0; i<n; i++){
        cin>>a[i];
    }
    int k; //找前k大的数
    cin>>k;

    arrangeRight(a, 0, n-1, k);

    //将数组右边k个数从小到大排序
    sort(a+n-k-1, a+n);

    for(int i=n-1; i>=n-k; i--){
        cout<<a[i]<<' ';
    }
    cout<<endl;

    return 0;
}