HDU - 1425 sort

给你n个整数，请按从大到小的顺序输出其中前m大的数。
Input
每组测试数据有两行，第一行有两个数n,m(0<n,m<1000000)，第二行包含n个各不相同，且都处于区间[-500000,500000]的整数。
Output
对每组测试数据按从大到小的顺序输出前m大的数。
Sample Input

5 3
3 -35 92 213 -644

Sample Output

213 92 3

Hint

请用VC/VC++提交

思路分析

方法一：使用快速排序的思想，判断比基准元素大的元素是否有k个，有则对后面的前k大个元素进行排序后输出前k大个元素，如果比基准元素大的元素超过k个，则在基准值的右边寻找前k大个元素，反之如果比基准元素大的元素少于k个，则在基准元素的左边寻找前k大个元素，直到基准元素及其后面有k个元素。
方法二：使用STL部分排序找出前k大个元素后输出。

代码

方法一

#include <iostream>
#include<algorithm>
#include<stdio.h>
using namespace std;
void msort(int a[],int s,int e,int n,int k){
    int temp=a[s];
    int i=s,j=e;
    while(i!=j){
        while(j>i&&a[j]>=temp)
            j--;//后面的元素比基准值小则退出循环
        a[i]=a[j];//小元素放到基准值之前
        while(i<j&&a[i]<=temp)
            i++;//前面的元素比基准值大则退出循环
        a[j]=a[i];//大元素放到基准值之后
    }
    a[i]=temp;
    if(n-i==k){
        sort(a+i,a+n,greater<int>());//对后面的k个元素进行排序
        while(i<n-1)
            cout<<a[i++]<<" ";//从i开始输出后面的k个元素（降序）
        cout<<a[i]<<endl;
    }else if(n-i>k){
        msort(a,i+1,e,n,k);//比a[s]大的元素超过k个，则在下标i的元素的右边寻找前k大的元素的下标
    }else{
        msort(a,s,i-1,n,k);比a[s]大的元素少于k个，则在下标i的元素的左边寻找前k大的元素的下标
    }
}

int main()
{
    int n,k,i;
    while(cin>>n>>k){
        int a[n+5];
        for(i=0;i<n;i++)
            scanf("%d",&a[i]);//注意这里不能用cin不然会超时
        msort(a,0,n-1,n,k);
    }
    return 0;
}

方法二

#include <iostream>
#include<algorithm>
#include<stdio.h>
using namespace std;
int main()
{
    int n,k,i;
    while(scanf("%d%d",&n,&k)!=EOF){
        int a[n+5];
        for(i=0;i<n;i++)
            scanf("%d",&a[i]);//注意这里不能用cin不然会超时
        partial_sort(a,a+k,a+n,greater<int>());//找出前k大个元素
        for(i=0,k--;i<k;i++)//输出k-1个元素
            cout<<a[i]<<" ";
        cout<<a[k]<<endl;//输出第k个元素
    }
    return 0;
}

注意事项

1. 有多组测试数据，所以要用while(cin>>n>>k)或while(scanf("%d%d",&n,&k)!=EOF)来执行多次。
2. 输入数组元素时，由于元素总数很大，所以要用scanf("%d",&a[i]);而不能用cin，不然会超时。