四.求一組數據中最大的K個數

和上篇求最小的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");
}