题目:
给定一个数组包含 n 个元素,统计前 m 个大的数并且把这 m 个数从大到小输出(m<n).
思路:
把前 m 大的都弄到数组最右边,然后对这最右边 m 个元素排序。
引入操作 FindMaxK(K): 把数组(或数组的一部分)前 K 大的都弄到最右边.
如何将前 K 大的都弄到最右边:
(1)设 key = a [0],将 key 挪到适当位置,
使得比 key 小的元素都在key左边,比 key 大的元素都在 key 右边(线性时间完成)
(2)选择数组的前部或后部再进行 FindMaxK 操作
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int a[100010];
void swap(int & a,int & b)
{
int tmp = a;
a = b;
b = tmp;
}
void FindMaxK(int a[],int s,int e,int k)
{
if( s - e +1 == k ){
return ;
}
int key = a[s];
int i = s,j = e;
while( i != j ) {
while( j > i && a[j] >= key ){
--j;
}
swap(a[i],a[j]);
while( i < j && a[i] <= key ){
++i;
}
swap(a[i],a[j]);
}
if( e-i+1 == k){
return ;
}else if( e-i+1>k ){
FindMaxK(a,i+1,e,k);
}else{
FindMaxK(a,s,i-1,k-e+i-1);
}
}
int main()
{
int n;
cin >> n;
for(int i = 0;i <n; ++i){
cin >> a[i];
}
int k;
cin >> k;
FindMaxK(a,0,n-1,k);
/*
for(int i = 0;i < n; ++i){
cout << a[i] << ",";
}
cout << endl;
*/
sort(a+n-k-1,a+n);
for(int i = n-1;i >= n-k; --i){
cout << a[i] << ",";
}
return 0;
}