POJ7617:输出前k大的数
输出前k大的数.
问题描述
- 给定一个数组,统计前k大的数并且把这k个数从大到小输出
思路
- 分治思想
- 把前k大的数移到最右边
1、设key=a[0],将key挪到适宜的位置,使得比key小的元素在key的左边,比前key大的元素在key的右边。
2、再选择数组的前部或者后部进行如上的排序。
代码
#include<bits/stdc++.h>
using namespace std;
int a[100001];
void f(int s,int e,int k){
if(e-s+1==k) return;
int i=s,j=e;
int key=a[s];
while(i!=j){
while(j>i&&a[j]>=key)
j--;
swap(a[i],a[j]);
while(i<j&&a[i]<key)
i++;
swap(a[j],a[i]);
}
if(e-i+1==k)
return;
else if(e-i+1>k)
f(i+1,e,k);
else if(e-i+1<k)
f(s,i-1,k-e+i-1);
}
int main(){
int n,k;
cin>>n;
for(int i=0;i<n;i++){
cin>>a[i];
}
cin>>k;
f(0,n-1,k);
sort(a+n-k,a+n);
for(int i=n-1;i>=n-k;i--)
cout<<a[i]<<endl;
return 0;
}