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

分治:把一个问题分解成形式和原任务相同,但规模更小的几个部分任务(通常是两个部分),分别完成,或只需要选一部分完成。然后再处理完成后的这一个或几个部分的结果,实现整个任务的完成。

把前m大的都弄到数组的右边,然后对最右边的m个元素排序再输出。
O(n)时间内实现把前m大的都弄到数组最右边。

arrangeRinght(int l, int r, int k):把数组[l, r]前k大的都弄到最右边。(类似快排)

#include<bits/stdc++.h>
using namespace std;

int a[100005], n;

void arrangeRinght(int l, int r, int k)
{
    int i=l, j=r;
    while(i<j)
    {
        while(i<j&&a[j]>=a[i])
            j--;
        swap(a[i], a[j]);
        while(i<j&&a[j]>=a[i])
            i++;
        swap(a[i], a[j]);
    }
    
    //分治
    if(r-i+1==k)
        return;
    if(r-i+1>k)
        arrangeRinght(i+1, r, k);//后半部分
    if(r-i+1<k)
        arrangeRinght(l, i-1, k-(r-i+1));//前半部分

}

int main()
{
    int k;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
    }
    scanf("%d",&k);
    arrangeRinght(1, n, k);
    sort(a+n-k, a+n+1);
    for(int i=n;i>=n-k+1;i--)
    {
        cout<<a[i]<<endl;
    }

    return 0;
}