找最大的N个数

题目：找最大的N个数

时间限制1000 ms，内存限制256000 kB，代码长度限制8000 B。

给定N个整数，找最大的前M个数(不一定有序)，要求时间复杂度尽量低。

先输入N和M (1 < = M < = N < = 50000)。然后下面N行每行代表一个整数（每个整数在int范围内），输出有M行，每行一个整数，表示最大的M个数。

输入示例：

5编程题-找最大的N个数

3

5

4

3

2

1

 输出示例：

5

4

3

分析：1、如果所有的数据都可以装入内存，可以考虑使用快排中的分割函数Partition()，找到前M个最大的数，此时的平均复杂度是O(N)。

[cpp]  view plain copy

1. void GetGreatestNum(int *input, int n, int*output, int k) {  
2.     if (!input || !output || n <= 0 || k <= 0 || k > n) return;  
3.     int start = 0;  
4.     int end = n - 1;  
5.     int index = Partition(input, n, start, end);  
6.     while (index != k - 1) {  
7.         if (index > k - 1)   
8.             end = index - 1;  
9.         else   
10.             start = index + 1;  
11.         index = Partition(input, n, start, end);  
12.     }  
13.       
14.     for (int i = 0; i < k; ++ i)   
15.         output[i] = input[i];  
16. }  
17.   
18. void Swap(int *a, int *b) {  
19.     int temp = *a; *a = *b; *b = temp;  
20. }  
21.   
22. int Partition(int *data, int length, int start, int end) {  
23.     // 大数在前，小数在后  
24.     if (data == NULL || length <= 0 || start < 0 || end >= n)   
25.         throw new std::exception("Invalid Parameters");  
26.     int index = RandomInRange(start, end);  
27.     Swap(&data[index], &data[end]);  
28.     int small = start - 1;  
29.     for (index = data[start]; index <= end; ++ index) {  
30.         if (data[index] > data[end]) {  
31.             ++ small;  
32.             if (small != index)  
33.                 Swap(&data[index], &data[small]);  
34.         }  
35.     }  
36.     ++ small;  
37.     Swap(&data[small], &data[end]);  
38.       
39.     return small;  
40. }  

2、如果是大数据的情况，要考虑使用小顶堆来实现，每次比较堆顶元素和新元素的大小，若堆顶元素小，交换两元素，重建最小堆。此时的复杂度为O(N log M)。STL容器中set、multiset是基于红黑树来实现的，可以直接用其来实现最小堆。

[cpp]  view plain copy

1. #include <stdio.h>  
2. #include <set>  
3.   
4. void GetGreatestNumber()  
5. {  
6.     FILE *file;  
7.     file = fopen("D:\\input.txt","r");  
8.     if (file == NULL)  
9.         return;  
10.     unsigned int N, M;  
11.     fscanf(file, "%d %d", &M, &N);  
12.     std::multiset<int, std::less<int> > minHeap;  
13.     for (unsigned int i = 0; i < M; ++ i)  
14.     {  
15.         int num;  
16.         fscanf(file,"%d", &num);  
17.         if (minHeap.size() < N)  
18.         {  
19.             minHeap.insert(num);  
20.         }  
21.         else  
22.         {  
23.             if (*(minHeap.begin()) < num)  
24.             {  
25.                 minHeap.erase(minHeap.begin());  
26.                 minHeap.insert(num);  
27.             }  
28.         }  
29.     }  
30.     std::multiset<int, std::less<int> >::reverse_iterator rIter;  
31.     for (rIter = minHeap.rbegin();  
32.         rIter != minHeap.rend();  
33.         ++ rIter)  
34.     {  
35.         printf("%d\n", *rIter);  
36.     }  
37.   
38.     fclose(file);  
39. }  

PS：有道，机试，2013，校招

参考：《剑指Offer》