分治专题 输出前k大的数

给定一个数组,统计前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;
}
全部评论

相关推荐

02-14 12:40
门头沟学院 Java
程序员花海:1.面试要求必须Java笔试不一定 2.难度对等秋招 远超于日常实习是因为同一批次且转正很多 竞争压力大 3.第一个加点指标,上线了就把接口性能加上去 使用本地缓存这个不算亮点 只是技术选型,要把为什么采用这个和背后的思考写出来而不是单纯堆叠技术没意义 4.八股要一直看 很容易忘记 5.拼团交易这个老问题 堆积技术 另外建议你把奖项合并到教育背景 没必要拆出来放最后
我的简历长这样
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务