题目描述
给定一个数组包含n个元素,统计前k大的数并且把这k个数从大到小输出。
输入
第一行是一个正整数n,表示数组的大小。n < 100000。
第二行包含n个整数,表示数组的元素,整数之间以一个空格分开。每个整数的绝对值不超过100000000。
第三行包含一个整数k。k < n。
输出
从大到小输出前k大的数。
样例输入
10
1 3 5 7 9 2 4 6 8 10
5
样例输出
10 9 8 7 6
思路1:排序后再输出。复杂度O(nlogn)
具体细节没什么好说的,直接调用algorithm中的sort();
思路2:分治。复杂度O(n+mlogm)
把前k大的数放到数组右边,然后对右边这k个数sort();
那如何在O(n)的时间内把前k大的数放到数组右边呢?想想快速排序吧 : )
代码
//输出前k大的数
#include <iostream>
#include <algorithm>
using namespace std;
void swap(int &a, int &b){
int tmp = a;
a = b;
b = tmp;
}
//在数组a中找前k大的数
void arrangeRight(int a[], int s, int e, int k){
if(s-e+1 == k){
return ;
}
//以第一个数为基准,大于base的放在数组右边,小的放左边
int base = a[s];
int i = s;
int j = e;
while(i != j){
while(i < j && a[j] >= base){
j--;
}
while(i < j && a[i] <= base){
i++;
}
if(i < j){
swap(a[i], a[j]);
}
}
//最后i==j,将其与第一个基准点交换
swap(a[i], a[s]);
//i右边个数
int right = e-i+1;
if(k == right){
return;
}
if(k < right){
arrangeRight(a, i+1, e, k);
}
if(k > right){
arrangeRight(a, s, i-1, k-right);
}
}
int main(){
int a[10];
int n; //共n个数
cin>>n;
for(int i=0; i<n; i++){
cin>>a[i];
}
int k; //找前k大的数
cin>>k;
arrangeRight(a, 0, n-1, k);
//将数组右边k个数从小到大排序
sort(a+n-k-1, a+n);
for(int i=n-1; i>=n-k; i--){
cout<<a[i]<<' ';
}
cout<<endl;
return 0;
}