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; 
}