/************************************/
//寻找在数组中最大的K个数
//小顶堆插入
void constructHeap(int a[],int n,int value){
a[n] = value;
int j,temp = value;
j = (n-1)/2;
while (j>=0 && n!=0)
{
if(a[j] <= temp)
break;
a[n] = a[j];
n = j;
j = (n-1)/2;
}
a[n] = temp;
}
//堆更新
void UpdataHeap(int a[],int index,int n)
{
int j,temp = a[index];
j = 2*index + 1;
while (j < n)
{
if(j+1 < n && a[j+1] < a[j])
++j;
if(a[j] >= temp)
break;
a[index] = a[j];
index = j;
j = index*2 +1;
}
a[index] = temp;
}
void MaxK(int a[],int n,int k){
int i;
int *temp = new int[k];
for (i=0;i<n;++i)
{
if(i < k)
constructHeap(temp,i,a[i]);//先构造k堆
else
{
if (temp[0] < a[i])
{
temp[0] = a[i];
UpdataHeap(temp,0,k);//更新k堆；
}
}
}
for (i=0;i<k;++i)
{
cout<<temp[i]<<" ";
}
delete []temp;
}
1. 建立大顶堆，维度为数组的个数，这里为20（第一次 插入的是每个数组中最大的值，即第一个元素）。
2. 删除最大堆堆顶，保存到数组或者栈中，然后向最大堆插入删除的元素所在数组的下一个元素。
3. 重复第1,2个步骤，直到删除个数为最大的K个数，这里为500.
/************************************/
//N个降序数组，找到最大的K个数
//大顶堆
void constructHeap1(int a[],int n,int value){
a[n] = value;
int j,temp = value;
j = (n-1)/2;
while (j>=0 && n!=0)
{
if(a[j] >= temp)
break;
a[n] = a[j];
n = j;
j = (n-1)/2;
}
a[n] = temp;
}
//堆更新
void UpdataHeap1(int a[],int index,int n)
{
int j,temp = a[index];
j = 2*index + 1;
while (j < n)
{
if(j+1 < n && a[j+1] > a[j])
++j;
if(a[j] <= temp)
break;
a[index] = a[j];
index = j;
j = index*2 +1;
}
a[index] = temp;
}
void GetMaxK(){
const int rows = 20;
const int cols = 500;
const int K = 500;
int data[rows][cols];
int result[K];//前500
int heap[20];//堆
int i,j;
for (i=0;i<rows;++i)
{
for (j=0;j<cols;++j )
{
data[i][j] = rand();
}
}
//快速排序
for (i = 0;i<rows;++i)
{
QuikSortDown(data[i],0,cols-1);
}
//开始
for(j=0;j<cols;++j)
{
for (i=0;i<rows;++i)
{
if(0 == j)
constructHeap1(heap,i,data[i][0]);//创建堆
else{
if((j-1)*rows + i >K-1 )
break;
result[(j-1)*rows + i] = heap[0];
heap[0] = data[i][j];
UpdataHeap1(heap,0,20);//更新堆
}
}
}
for (i=0;i<K;++i)
{
cout<<result[i]<<" "<<endl;
}
}原文地址：http://blog.csdn.net/h_wlyfw/article/details/38925625
