这题采用分治的思想,很奇怪,我试着直接把整个数组进行排序,然和直接输出结果居然也没超时。。。。。这里最好的办法是用分治,减少时间。先将第一个数s进行排列,使得比s小的都在s的左边,同时比s大的都在s的右边,然后假设从s到最后的数(包括s)是range个,如果range==k,那就return了,把最后k个数快速排序后输出就好了。如果range<k,就再从s的左边重复进行上述操作,但这次你只需要k-rang个数,如果range>k,就从s的右边进行查找,仍然需要找k个数,然后一直下去一直找就ok了。
#include<iostream>
using namespace std;
void Swap(int & a, int & b)
{
int temp = a;
a = b;
b = temp;
}
void QuickSort(int a[], int start, int end)//快速排序
{
if(start >= end) return;
int i= start;
int j= end;
while( i != j)
{
while( j > i && a[j] >= a[i])
j--;
Swap(a[i],a[j]);
while( i < j && a[i] <= a[j])
++i;
Swap(a[i],a[j]);
}
//注意下面的范围,不然很容易出错
QuickSort(a,start,i-1);
QuickSort(a,i+1,end);
}
void Find(int a[],int start,int end,int k)
{
int s = a[start];
int i=start;
int j=end;
//快速排序的思想
while(i != j)
{
while( i<j && s <= a[j])
j--;
Swap(a[i],a[j]);
while( i<j && a[i] <= s)
i++;
Swap(a[i],a[j]);
}
int range = end - i + 1;
if(range == k) return;
//同样注意范围
if(range < k) Find(a,start,i-1,k-range);
if(range > k) Find(a,i+1,end,k);
}
int main()
{
int n,k;
cin>>n;
int a[n];
for(int i = 0; i < n; ++i)
{
cin>>a[i];
}
cin>>k;
Find(a,0,n-1,k);
QuickSort(a,n-k,n-1);
for(int i = n-1; i >= n-k; --i)
{
cout<<a[i]<<endl;
}
return 0;
}