/******************************************************
尋找k個最小的數

方法：堆排序

***************************************************/





#include "stdafx.h"
#include<iostream>
#include<assert.h>
using namespace std;

void MaxHeapAdjust(int a[],int i,int n)//對節點i進行調整,n是陣列元素個數  
{   
    int largest = 0;  
    int left = i*2 + 1;//節點i的左節點  
    int right = i*2 + 2;//節點i的右節點  
      
    if(left< n && a[left]>a[i])//左節點和父節點比較  
        largest =left;  
    else  
        largest = i;  
    if(right<n && a[right]>a[largest])//如果有右節點，且左小於右  
        largest = right;  
          
    if(i != largest)//如果最大節點不是父節點  
    {  
        swap(a[i],a[largest]);  
        MaxHeapAdjust(a,largest,n);//下降父節點，跟子樹最大堆比，其實經過建立最大堆時，子樹都為最大堆了  
    }     
          
  
}  
  
  
//建立最大堆  
void BuildMaxHeap(int a[],int length)  
{  
    for(int i = length/2-1;i>=0;i--)//從非葉子節點開始進行最大堆調整  
    {  
        MaxHeapAdjust(a,i,length);  
    }  
  
}  
int main()
{
	//定義陣列儲存堆元素
	int k;
	cout << "請輸入k：" << endl;  
    cin >> k;  
	int length = 0;  
    int i = 0; 
	int num = 0;
    int a[100] = {0};  
  
    cout << "請輸入前k個數：" << endl;  
    for(;i<k;i++)  
        cin>>a[i]; 
    BuildMaxHeap(a,k);            //建堆
    cout << "請繼續輸入數字:" << endl;
	 while(cin >>num)                              //ctrl+z結束
		  {
		<span style="white-space:pre">	</span>if(num < a[0])
			{
			    a[0] = num;
			    MaxHeapAdjust(a,0,k);
			}
		 }
 

	
	 for(int i = 0;i<k;i++)  
    {  
        cout << a[i]<<" ";  
    }  
	 cout << endl;

	
	 system("pause");
	 return 0;

}
 

用STL的multiset來實現

typedef multiset<int,greater<int> > iSet;//set中的元素從大到小排列
typedef multiset<int,greater<int> >::iterator setIter;

void GetLeastKnum(const vector<int>& data,iSet & iset,int k)
{
	iset.clear();
	if(k<1 || data.size()<k)
		return;
	//遍歷陣列，不斷更新堆

	vector<int>::iterator iter = data.begin();
	for(;iter!=ivec.size();iter++)
	{
		if(iset.size()<K)
			iset.insert(*iter);
		else
		{
			iSet siterGreatst = iset.begin();
			if(*iter < *siterGreatst)
				{
					iSet.erase(siterGreatst);
					iSet.insert(*iter);
				}
		}
	}
}