001:输出前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

分治思想,抠细节较多,所以看代码吧

#include<cstdio>
#include<algorithm>
using namespace std;
void Swap(int & a,int & b)
{
    int tmp = a;
    a = b;
    b = tmp;
}
int k,n;
//移动,并返回基准值的下标
int GetIndex(int a[],int s,int e)
{
    int k = a[s]; //选第一个元素为基准元素
    int i = s,j = e;
    while(i!=j) {
        while(j>i && a[j]>=k)
            --j;
        Swap(a[i],a[j]);
        while(j>i && a[i]<=k)
            ++i;
        Swap(a[i],a[j]);
    }
    return i;
}
//找第k大的数,并且把比该元素大的全部移到右边
void  FindKth(int a[],int s,int e,int k)
{
    int pos = GetIndex(a,s,e);//基准值的位置
    int cnt = e-pos+1;//为a[pos,e]的个数
    if(cnt== k) return; //个数为k说明找完
    else if(cnt > k) FindKth(a,pos+1,e,k); //右边元素多了,继续
    else FindKth(a,s,pos-1,k-cnt); //右边元素少了,那么得从左边找k-cnt个元素,也就是找前k-cnt大的元素
}
int main()
{
    scanf("%d",&n);
    int a[100005];
    for(int i = 0; i<n ; ++i)
        scanf("%d",&a[i]);
    scanf("%d",&k);
    FindKth(a,0,n-1,k);
    sort(a+n-k,a+n);
    for(int i = n-1; i>=n-k; --i)
        printf("%d\n",a[i]);
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值