和上篇求最小的K个数有所区别的是,本篇用的是最小堆,其他类似。
时间复杂度同样为n*logK;
#include<iostream>
using namespace std;
void adjustHeap(int b[],int s,int k)
{
int temp=b[s];
for(int i=2*s+1;i<=k;i=i*2+1)
{
if((i+1)<=k&&b[i]>b[i+1])
i++;
if(temp<b[i])
break;
b[s]=b[i];
s=i;
}
b[s]=temp;
}
void buildMinHeap(int b[],int k)
{
for(int i=k/2-1;i>=0;i--)
{
adjustHeap(b,i,k-1);
}
cout<<"建堆后的b:";
for(int i=0;i<k;i++)
{
cout<<b[i]<<" ";
}
cout<<endl;
}
void main()
{
int a[]={2,7,4,9,0,1,7,5,10,55,76,15};
int n=sizeof(a)/sizeof(int);
int k;
cout<<"请输入最小要选择最小数的个数"<<endl;
cin>>k;
int *b=(int *)malloc(sizeof(int)*k);
for(int i=0;i<k;i++)
{
b[i]=a[i];
}
buildMinHeap(b,k);
for(int i=k;i<n;i++)
{
if(a[i]>b[0])
{
b[0]=a[i];
adjustHeap(b,0,k-1);
}
}
for(int i=0;i<k;i++)
{
cout<<b[i]<<" ";
}
cout<<endl;
system("pause");
}