用堆排序实现查找最小的K个元素

转载自：http://blog.chinaunix.net/uid-28977986-id-3864505.html

  2013-08-24 19:25:51

分类： C/C++

题目：查找最小的k个元素。

利用堆排序解决这个问题的思路有两种：
   思路 1：将输入的n个数建立成大顶堆并排序后，再遍历输出前k个元素，此k个元素即使最小的K个元素。
  思路2:  维护k个元素的最大堆，即用容量为k的最大堆存储最先遍历到的k个数，并假设他们是最小的k个数。继续遍历数列，每次遍历到一个元素，将它与堆顶元素比较，若小于堆顶元素，更新堆，否则，不更新堆。

　思路１的代码和实现堆排序的代码相同。
　
  思路2的代码实现如下：
   

1. #include<iostream>
   
2. using namespace std;
   
3. 
   
4.  void swap(int &a,int &b)
   
5.  {
   
6.      int tmp;
   
7.      tmp=a;
   
8.      a=b;
   
9.      b=tmp;
   
10.  
    
11.  }
    
12. 
    
13. 
    
14. void HeapAdjust(int a[],int i,int k)
    
15. {
    
16.     int lchild,rchild;
    
17.     int max=i;
    
18.     lchild=2*i;
    
19.     rchild=2*i+1;
    
20.     if(i<=k/2)
    
21.     {
    
22.         if(lchild<=k && a[lchild]>a[max])
    
23.          max=lchild;
    
24.      if(rchild<=k && a[rchild]>a[max])
    
25.          max=rchild;
    
26.      if(max!=i)
    
27.         {
    
28.          swap(a[max],a[i]);
    
29.      HeapAdjust(a,max,k);
    
30.         }
    
31.     }
    
32. 
    
33. }
    
34. 
    
35. 
    
36. int main()
    
37. {
    
38.     int i,j,k,size;
    
39.     int a[100];
    
40.     printf("please input the size of heap:");
    
41.     while(scanf("%d",&size)==1 && size>0)
    
42.     {
    
43.         printf("请输入堆的元素值：");
    
44.      for(i=1;i<=size;i++)
    
45.             scanf("%d",&a[i]);
    
46.         printf("请输入ｋ的值");
    
47.         scanf("%d",&k);
    
48. 
    
49.         for(i=k/2;i>=1;i--)
    
50.             HeapAdjust(a,i,k);
    
51.         
    
52.      for(j=k+1;j<=size;j++)
    
53.         {
    
54.          if(a[j]<a[1])
    
55.             {
    
56.              a[1]=a[j];
    
57.                 HeapAdjust(a,1,k);
    
58.             
    
59.             }
    
60.         }
    
61. 
    
62.         for(i=1;i<=k;i++)
    
63.             cout<<a[i]<<" ";
    
64.     }
    
65.     return 0;
    
66. 
    
67. 
    
68. }