最小的K个数,堆维护问题

void AdjustDown(double A[], int i, int len)  
{  
	double temp = A[i];  // 暂存A[i]  

	for(int largest=2*i+1; largest<len; largest=2*largest+1)  
	{  
		if(largest!=len-1 && A[largest+1]>A[largest])  
			++largest;         // 如果右子结点大  
		if(temp < A[largest])  
		{  
			A[i] = A[largest];  
			i = largest;         // 记录交换后的位置  
		}  
		else  
			break;  
	}  
	A[i] = temp;    // 被筛选结点的值放入最终位置  
}

/* 建堆 */
void BuildMaxHeap(double A[], int len)
{
	for(int i=len/2-1; i>=0; --i)  // 从i=n/2-1到0,反复调整堆
		AdjustDown(A, i, len);
}


/* 维护 A[0...k-1] 这个大根堆 */
void topK(double A[], int n, int k)
{
	BuildMaxHeap(A, k);  // 先用前面的k个数建大根堆
	for(int i=k; i<n; ++i)
	{
		if(A[i] < A[0])  // 如果小于堆顶元素,替换之
		{
			double tmp = A[0];
			A[0] = A[i];
			A[i] = tmp;
			AdjustDown(A, 0, k);  // 向下调整
		}
	}
}

原文地址:https://songlee24.github.io/2015/03/21/hua-wei-OJ2051/

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值