描述
给定一个数组包含n个元素,统计前m大的数并且把这m个数从大到小
输出。
输入
第一行包含一个整数n,表示数组的大小。n < 100000。
第二行包含n个整数,表示数组的元素,整数之间以一个空格分开
。每个整数的绝对值不超过100000000。
第三行包含一个整数m。m < n。
输出
从大到小输出前m大的数,每个数一行。
#include <iostream>
using namespace std;
const int MAXN = 100010;
int a[MAXN];
void swap(int &a,int &b){
int tmp = a;
a = b;
b = tmp;
}
void arrangeRight(int a[],int s,int e,int m){
int key = a[s];
int i = s,j = e;
while(i != j){
while(i < j && a[j] >= key)
j--;
swap(a[i],a[j]);
while(i < j && a[i] <= key)
i++;
swap(a[i],a[j]);
}
//此时a[i]即为key,key左边的数都比他小,右边的数都比他大
int now = e - i + 1;
if(now == m)
return;
if(now > m)
arrangeRight(a,i+1,e,m);
else
arrangeRight(a,s,i-1,m - now);
}
void QuickSort(int a[],int s,int e){
if(s >= e)
return;
int k = a[s];
int i = s,j = e;
while(i != j){
while(i < j && a[j] >= k)
j--;
swap(a[i],a[j]);
while(i < j && a[i] <= k)
i++;
swap(a[i],a[j]);
}
QuickSort(a,s,i-1);
QuickSort(a,i+1,e);
}
int main(){
int n,m;
cin >> n;
for(int i = 0;i<n;i++) cin >> a[i];
cin >> m;
arrangeRight(a,0,n-1,m);
QuickSort(a,n-m,n-1);
for(int i = n-1;i>=(n-m);i--)
cout << a[i] << " ";
return 0;
}