「数据结构详解·十四」对顶堆

Abstrummerain

已于 2024-07-04 06:56:07 修改

阅读量986

点赞数 8

分类专栏：「数据结构详解」文章标签：数据结构 c++ 算法

于 2024-07-04 06:51:42 首次发布

本文链接：https://blog.csdn.net/leo_chenjy/article/details/140162755

版权

「数据结构详解」专栏收录该内容

14 篇文章 3 订阅

订阅专栏

1. 对顶堆的概念

对顶堆，是二叉堆的一种应用。它是将处理的数据，分成一个小根堆、一个大根堆，然后可以快速求解出一些东西。
下面我们直接通过例题来理解。

2. 例题详解

2-1. P1168 中位数

我们可以开一个大根堆 $q_{\max}$ ，维护较小的值；再开一个小根堆 $q_{\min}$ ，维护较大的值。
每次加入一个数 $x$ ：若 $x$ 大于大根堆堆顶 $t_{\max}$ ，则加入小根堆；若 $\le t_{\max}$ ，则加入大根堆。
为什么要这么做呢？
首先，这样可以保证大根堆的元素值都小于小根堆的元素值。
那么这样的话，对于 $t_{\max}$ ，有 $q_{\min}|$ 个元素大于 $t_{\max}$ ，有 $q_{\max}|-1$ 个元素小于等于 $t_{\max}$ ；对于 $t_{\min}$ 是正好相反的。
那么我们要找到当前中位数，只要让 $||q_{\max}|-|q_{\min}||\le1$ （即两个堆的元素数差值不大于）即可。
至于怎么做，不难想到，只要元素个数多的堆的堆顶取出，放入元素个数少的堆即可。然后奇数项的中位数就是元素个数多的堆的堆顶。
而时间复杂度，对于查询操作，自然是 $O (1)$ 的；而每次插入元素后很难不发现最多调整一次，因此总时间复杂度是 $O(N\log N)$ 。
下面我们来模拟一下样例二。
对于下面的示意图，上面是 $q_{\min}$ ，下面是 $q_{\max}$ 。

首先遇到 $3$ ，大根堆为空，加入 $q_{\max}$ 。

此时中位数为 $t_{\max}=3$ 。

接着遇到 $1$ ， $1\le t_{\max}=3$ ，故加入 $q_{\max}$ 。

注意到 $q_{\max}|-|q_{\min}|=2>1$ ，因此将 $q_{\max}$ 的堆顶元素调整到 $q_{\min}$ 中。

然后遇到 $5$ ， $5> t_{\max}=1$ ，故加入 $q_{\min}$ 。

此时中位数为 $t_{\min}=3$ 。

紧接着遇到 $9$ ， $9> t_{\max}=1$ ，故加入 $q_{\min}$ 。

注意到 $q_{\min}|-|q_{\max}|=2>1$ ，因此将 $q_{\min}$ 的堆顶元素调整到 $q_{\max}$ 中。

再下来遇到 $8$ ， $9> t_{\max}=3$ ，故加入 $q_{\min}$ 。

此时中位数为 $t_{\min}=5$ 。

再接下来遇到 $7$ ， $7> t_{\max}=3$ ，故加入 $q_{\min}$ 。

注意到 $q_{\min}|-|q_{\max}|=2>1$ ，因此将 $q_{\min}$ 的堆顶元素调整到 $q_{\max}$ 中。

最后遇到 $6$ ， $6> t_{\max}=5$ ，故加入 $q_{\min}$ 。

此时中位数为 $t_{\min}=6$ 。

参考代码：

#include<bits/stdc++.h>
using namespace std;

priority_queue<int>a;
priority_queue<int,vector<int>,greater<int>>b;

int read()
{
	int s=0,f=1;
	char ch=getchar();
	while(ch<'0'||ch>'9')
	{
		if(ch=='-') f=-1;
		ch=getchar();
	}
	while(ch>='0'&&ch<='9')
	{
		s=s*10+ch-48;
		ch=getchar();
	}
	return s*f;
}

int main()
{
	int n=read();
	a.push(read());
	printf("%d\n",a.top());
	for(int i=2;i<=n;i++)
	{
		int m=read();
		if(m>a.top()) b.push(m);
		else a.push(m);
		while(abs((int)a.size()-(int)b.size())>1)
		{
			if(a.size()>b.size())
			{
				b.push(a.top());
				a.pop();
			}
			else
			{
				a.push(b.top());
				b.pop();
			}
		}
		if(i&1) printf("%d\n",a.size()>b.size()?a.top():b.top());
	}
	return 0;
}