其他分享
首页 > 其他分享> > 大根堆、小根堆的应用—找中位数、O(logn)实现(你是不是只会排序呀,还不快点进来看看)

大根堆、小根堆的应用—找中位数、O(logn)实现(你是不是只会排序呀,还不快点进来看看)

作者:互联网

1、思路步骤:

 step:   1)先从用户获得一个数据,放在大根堆;
            2)在获得一个数据与大根堆的堆顶进行比较,若小于等于堆顶就放入大根堆,否则                    放入小根堆;
            3)再比较大根堆的size和小根堆的size,若两者相差超过2,就将size较大的那个的栈                     顶弹出放入size较小的堆中;
            4)循环往复,直到遍历完所有数字,若最后两堆的size不相等,就将size较大的那个                   堆顶 弹出个;否者都弹出取平均数。

2.代码实现: 

int findMedian(int arr[], int sz) {
	if (sz == 0)
		return arr[0];
	priority_queue<int,queue<int>,greater<int>> bigQ;//大根堆
	priority_queue<int,queue<int>,less<int>> smallQ;//小根堆
	bigQ.push(arr[0]);
	int bSz, sSz;
	for (int i = 1;i < sz;i++) {
		if (arr[i] <= bigQ.top()) //若小于等于堆顶就放入大根堆
			bigQ.push(arr[i]);
		else//否则放入小根堆
			smallQ.push(arr[i]);
		bSz = bigQ.size();//拿到大根堆的大小
		sSz = smallQ.size();//拿到小根堆的大小
		if (fabs(bSz - sSz) > 1) {
			if (bSz > sSz) {//就将size较大的那个的栈顶弹出放入size较小的堆中
				smallQ.push(bigQ.top());
				bigQ.pop();
			}
			else {
				bigQ.push(smallQ.top());
				smallQ.pop();
			}
		}
	}
	bSz = bigQ.size();
	sSz = smallQ.size();
	if (bSz > sSz)
		return bigQ.top();
	if (bSz < sSz)
		return smallQ.top();
	if (bSz = sSz)
		return (bigQ.top() + smallQ.top()) / 2;
}

标签:smallQ,大根堆,bSz,sSz,top,bigQ,logn,小根堆,size
来源: https://blog.csdn.net/m0_58086930/article/details/120383348