【分治】输出前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

排序后再输出，复杂度 O(nlogn)
用分治处理： 复杂度 O(n+mlogm)
思路：把前m大的都弄到数组最右边，然后对这最右边m个元素排序，
再输出
关键 ： 利用快排思想在O(n)时间内实现把前m大的都弄到数组最右边 

#include<iostream>
using namespace std;
int n;
int a[100010];
int k;
int partition(int s, int e)
{
    int key = a[s];
    int i = s;
    int j = e;
    while (i < j)
    {
        while (a[j] >= key & i < j)
        {
            j--;
        }
        while (a[i] <= key & i < j)
        {
            i++;
        }
        swap(a[i], a[j]);
    }
    if (a[i] < key)
    {
        swap(a[s], a[i]);
    }
    else
        swap(a[s], a[i - 1]);
    return i;
}
void quickSort(int s,int e)
{
    if (s < e)
    {
        int m = partition(s, e);
        quickSort(s, m - 1);
        quickSort(m+1, e);
    }
}
void arrangeRight(int s,int e,int k)
{
    if (e - s + 1 == k)//如果要找到大数个数就是数组长度，不进行操作，直接排序
    {
        return;
    }
    //找一个基准数，使比这个数大的都在右边，小的都在左边
    int key = a[s];
    int i = s;
    int j = e;
    while (i < j)
    {
        while ((a[j] >= key) && (i < j))
        {
            j--;
        }
        while ((a[i] <= key) &&( i < j))
        {
            i++;
        }
        swap(a[i], a[j]);
    }
    if (a[i] < key)
    {
        swap(a[s], a[i]);
    }
    else
    {
        swap(a[s],a[i-1]);
    }
    int index = i;//最后基准数的位置
    int len = e - index + 1;
    if (len == k)
    {
        return;
    }
    else if (len > k)//找到的大数多余规定的，再往右缩小范围
    {
        arrangeRight(index+1,e,k);
    }
    else//找到的大数不够k个，往左扩大寻找范围，已找到len个，还差len-m个
    {
        arrangeRight(s,index-1,k-len);
    }
}
int main()
{
    cin >> n;
    for (int i = 0; i < n; ++i)
    {
        cin >> a[i];
    }
    cin >> k;
    arrangeRight(0, n - 1,k);
    quickSort(n-k,n-1);
    for (int i = n-1; i >= n - k; --i)
    {
        cout << a[i] << endl;
    }
    system("pause");
    return 0;
}