从数组中找出最小的k个数

//调整堆(小顶堆) 
void HeapAdjust(int A[],int parent,int length) 
{
	//temp保存当前父节点parent
	int temp=A[parent];
	
	//左子结点开始,i节点的左孩子为2i+1,右孩子为2i+2
	for(int child=2*parent+1;child<length;child=2*child+1)
	{
		//如果右子节点存在,且右子结点小于左子结点,则指向右子结点
		if(child+1<length && A[child]>A[child+1])
		    child++;
		
		//如果(左或右)子节点小于父节点,将子节点值赋给父节点(不用进行交换)
		if(A[child]<temp)
		{
			A[parent]=A[child];
			parent=child;//选取孩子结点的左孩子结点,child=2*child+1,继续向下筛选
		}
		else //若父节点小于(左和右)孩子节点 ,则直接结束
			break;
	}	
		A[child]=temp;//将temp的值放到最终的位置 
}

//建立堆 
void HeapBuild(int A[],int length)
{
	for(int i=length/2-1;i>=0;i--)
	    HeapAdjust(A,i,length)
}

//利用堆找出最小的k个数(小顶堆) 
int getKMinusByHeap(int read[],int k)
{
	if(k<1 || k>read.length())
	    return -1;
	
	int kHeap = new int[k]; //存储k个堆元素 
	
	//初始时一次性从文件中读取k个数据到堆中 
	for(int i=0;i<k;i++)
	    kHeap[i]=read[i];
	
	HeapBuild(kHeap,k); //建小顶堆堆,时间复杂度O(k)
	
	// 从文件中一个一个的读取剩余数据
	for(int i=k;i<read.length();i++)
	{
		if(read[i]<kHeap[0]) //读入的元素小于堆顶元素
		{
			kHeap[0]=read[i];
			HeapAdjust(kHeap,0,k);// 从堆顶开始向下进行调整,时间复杂度O(logk)
		} 
	}
	return kHeap; 
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值