top K 问题

（1）寻找大富翁问题。

题目描述：

    浙江桐乡乌镇共有n个人,请找出该镇上的前m个大富翁.

输入：

    输入包含多组测试用例.
    每个用例首先包含2个整数n（0<n<=100000）和m(0<m<=10)，其中: n为镇上的人数，m为需要找出的大富翁数, 接下来一行输入镇上n个人的财富值.
    n和m同时为0时表示输入结束.

输出：

    请输出乌镇前m个大富翁的财产数，财产多的排前面，如果大富翁不足m个，则全部输出,每组输出占一行.

这其实是一个Top k问题，求一个大数组中的最大的k个数。

（2）算法实现思想：使用最小堆来寻找最大的k个数，并且使用快速排序对最小堆进行排序，按从大到小的顺序输出。

[cpp]  view plain copy

1. #include<iostream>  
2.   
3. using namespace std;  
4.   
5. void quick_sort(int a[], int left, int right) {  
6.     int temp = a[left];  
7.     int i = left, j = right;  
8.     if (left < right) {  
9.         while (i < j) {  
10.             while (a[j] > temp && i < j)  
11.                 j--;  
12.             if (i < j)  
13.                 a[i++] = a[j];  
14.             while (a[i] < temp && i < j)  
15.                 i++;  
16.             if (i < j)  
17.                 a[j--] = a[i];  
18.         }  
19.         a[i] = temp;  
20.         quick_sort(a, left, i - 1);  
21.         quick_sort(a, i + 1, right);  
22.     }  
23. }  
24. //修整最小堆  
25. void fix_min_heap(int a[], int pos, int len) {  
26.     int left = 2 * pos + 1;  
27.     int right = 2 * pos + 2;  
28.     int min = pos;  
29.     if (left < len && a[left] < a[pos])  
30.         min = left;  
31.     if (right < len && a[right] < a[min])  
32.         min = right;  
33.     if (pos != min) {  
34.         int tep = a[pos];  
35.         a[pos] = a[min];  
36.         a[min] = tep;  
37.         fix_min_heap(a, min, len);  
38.     }  
39. }  
40. //建立最小堆  
41. void build_min_heap(int a[], int len) {  
42.     int parent = (len - 1) / 2;  
43.     int i = 0, j = 0;  
44.     for (i = parent; i >= 0; --i) {  
45.         int left = 2 * i + 1;  
46.         int right = 2 * i + 2;  
47.         int min = i;  
48.         if (left < len && a[left] < a[i])  
49.             min = left;  
50.         if (right < len && a[right] < a[min])  
51.             min = right;  
52.         if (i != min) {  
53.             int tep = a[i];  
54.             a[i] = a[min];  
55.             a[min] = tep;  
56.             fix_min_heap(a, min, len);  
57.         }  
58.     }  
59. }  
60.   
61. int main() {  
62.     int total[100000];  
63.     int min_heap[10];  
64.     int n, m, i=0, j=0;  
65.     while(cin>>n>>m)  
66.     {  
67.         if(m == 0&& n == 0)  
68.             break;  
69.         for(i=0; i<n; i++)  
70.             cin>>total[i];  
71.         for(j=0;j<m; j++)  
72.             min_heap[j] = total[j];  
73.         build_min_heap(min_heap,  m);  
74.         for(i=m; i<n; i++)  
75.         {  
76.             if(min_heap[0]<total[i])  
77.             {  
78.                 min_heap[0] = total[i];  
79.                 fix_min_heap(min_heap, 0, m);  
80.             }  
81.         }  
82.         quick_sort(min_heap,  0, m-1);  
83.         for(i=m-1; i>0; --i)  
84.             cout<<min_heap[i]<<" ";  
85.         cout<<min_heap[0]<<endl;  
86.     }  
87. }