010:输出前k大的数
总Time Limit: 10000ms 单个测试点时间限制: 1000ms Memory Limit: 65536kB
Description
给定一个数组,统计前k大的数并且把这k个数从大到小输出。
Input
第一行包含一个整数n,表示数组的大小。n < 100000。
第二行包含n个整数,表示数组的元素,整数之间以一个空格分开。每个整数的绝对值不超过100000000。
第三行包含一个整数k。k < n。
Output
从大到小输出前k大的数,每个数一行。
Sample Input
10
4 5 6 9 8 7 1 2 3 0
5
Sample Output
9
8
7
6
5
解题思路:
排序后再输出,复杂度 O(nlogn)
用分治处理:复杂度 O(n+mlogm)
思路:把前m大的都弄到数组最右边,然后对这最右边m个元素排序, 再输出
关键 :O(n)时间内实现把前m大的都弄到数组最右边
引入操作 arrangeRight(k): 把数组(或数组的一部分)前k大的 都弄到最右边
如何将前k大的都弄到最右边
1)设key=a[0], 将key挪到适当位置,使得比key小的元素都在 key左边,比key大的元素都在key右边(线性时间完成)
2) 选择数组的前部或后部再进行 arrangeRight操作
a = k ————done
a > k ————对最右边a-1个元素再进行arrangeRigth(k)
a < k ————对左边b个元素再进行arrangeRight(k-a)
将前m大的都弄到数组最右边的时间:
T(n) = T(n/2) + a*n
= T(n/4) + a*n/2 + a*n
= T(n/8) + a*n/4 + a*n/2 + a*n
= ...
= T(1) + ... + a*n/8 + a*n/4 + a*n/2 + a*n
< 2*a*n
即 O(n)
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int a[100010];
void arrangeRight(int s,int e,int k) //把数组的前 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 ) //后面的元素比 k 大:符合情况
--j;
swap(a[i],a[j]); //正是因为找到了后面的元素比 K 小了,所以要进行置换
while( i < j && a[i] <= key ) //前面的元素比 k 小:符合情况
++i;
swap(a[i],a[j]); //正是因为找到了前面的元素比 k 大了,所以要进行置换
} //处理完后,a[i] = key
int sum = e+1 - i;//数组中从key开始往右的元素个数和
if(sum == k) //完成
return ;
else if(sum > k) //对最右边的 sum-1 个元素再进行arrangeRight(k)
arrangeRight(i+1,e,k);
else if(sum < k)
arrangeRight(s,i-1,k-sum); // 对左边 b(key之前的元素个数) 个元素再进行arrangeRight(k-sum)
}
int main()
{
int n,k;
scanf("%d",&n);
for(int i = 0; i<n ;i++)
{
scanf("%d",&a[i]);
}
scanf("%d",&k);
arrangeRight(0,n-1,k); //把数组的前 k 大的元素都弄到最右边
sort(a+n-k,a+n);
for(int i = n-1; i >= n-k ; --i) //输出
cout << a[i] << endl;
return 0;
}