编程之美——寻找最大的K个数（最小堆实现）

// 寻找最大的k个数（堆）.cpp : 定义控制台应用程序的入口点。

//

#include "stdafx.h"

#include<iostream>

using namespace std;

static const int N=100;

void MinHeapify(int origin[],int i,int heapsize);

void BuildMinHeap(int origin[],int heapsize);

int _tmain(int argc, _TCHAR* argv[])

{

int origin
;

int k;

cout<<"origin array:"<<endl;

for(int i=0;i<N;i++)

{

origin[i]=rand()%(N+1);

cout<<origin[i]<<" ";

}

cout<<endl<<"请输入你想找最大数的个数: ";

cin>>k;

BuildMinHeap(origin,k);

for(int i=k;i<N;i++)

{

if(origin[i]>origin[0])

{

swap(origin[0],origin[i]);

MinHeapify(origin,0,k);

}

}

for(int j=0;j<N;j++)

{

cout<<origin[j]<<" ";

}

return 0;

}

void MinHeapify(int origin[],int i,int heapsize)//保持最小堆的性质

{

int smallest;

int l=2*i+1;

int r=2*i+2;

if(l<heapsize&&origin[i]>origin[l])

{

smallest=l;

}

else

{

smallest=i;

}

if(r<heapsize&&origin[r]<origin[smallest])

{

smallest=r;

}

if(smallest!=i)

{

swap(origin[i],origin[smallest]);

MinHeapify(origin,smallest,heapsize);

}

}

void BuildMinHeap(int origin[],int heapsize)//构建最小堆

{

for(int i=heapsize/2-1;i>=0;i--)

{

MinHeapify(origin,i,heapsize);

}

}