堆

堆

• 堆总是满足下列性质

堆中某个结点的值总是不大于（或不小于）其父节点的值
堆总是一棵完全二叉树

• 完全二叉树

若设二叉树的深度为h，除了第h层以外，其他各层的结点数都达到最大个数，第h层所有的节点都连续集中在左边。

• 关于堆，最重要的操作是插入（push），和弹出（pop），他们的算法复杂度都是O(logn)。（以最小堆为例）
• 对于插入来说，首先在堆的末尾插入该数值，然后不断向上提升直到没有大小颠倒为止。
• 对于删除来说，首先删除最小值，其次将末尾结点的值复制到根结点上，然后不断向下交换直到没有大小颠倒为止。
• 对于二叉树，除了使用链表以外，还可以使用数组，设某结点的数组下标为n，则其左子结点的下标为2n+1，其右子结点的下标为2n+2。
• 下面是用数组实现最小堆的插入和删除操作的代码

int heap[MAX_N], sz = 0;	//声明堆数组，结点个数
void push(int x){
	int i = sz++;	//首先获取自己结点的编号
	while (i > 0){
		int p = (i - 1) / 2;	//获取父结点位置编号
		if (heap[p] <= x) break;	//如果大小没有颠倒，退出
		//交换（将父结点的值赋给自己）
		heap[i] = heap[p];	
		i = p;
	}
	heap[i] = x;	//最后将要插入的值放入找出的位置
}
int pop(){
	//把根结点（最小值）返回
	int ret = heap[0];
	int x = heap[--sz];	//要提到根的数值（最后一位）
	int i = 0;
	while(i * 2 + 1 < sz){	//从根结点开始向下比较
		//比较儿子的值，选择小的那个
		int left = 2 * i + 1, right = 2 * i + 2;
		if (right < sz && heap[right] < heap[left]) left = right;
		//如果没有大小颠倒就退出
		if (heap[left] >= x) break;
		//把儿子的值提上来
		heap[i] = heap[left];
		i = left; 
	}
	heap[i] = x;
	return ret;
}

另一种理解方法

• 其实堆中最本质的操作只有两种，一种是上移，另一种是下移。而为了实现堆的操作，只需要进行相应的上移和下移操作，就可以实现堆的基本功能
  假设上移操作为up，下移操作为down

1. push：将值插入到尾结点，然后up尾结点
2. pop：将尾节点数值复制到根节点上，删除尾节点，然后down根节点
3. top：访问根节点heap[1]
4. 删除某个值：将尾节点的值复制到当前节点，删除当前节点，对当前节点进行up和down操作
5. 修改某个值：修改当前节点的值，对当前节点进行up和down操作

• 4和5为用数组实现的堆比stl提供的优先队列所多的功能，为了不人工进行比较大小操作，所以同时进行上移和下移操作，因为两种操作只会进行一个，所以时间复杂度没有增加

int heap[N], sz = 0;
void down(int u){
	int t = u;
	//判断两个子节点是否在堆里，并且找出本节点与两个子节点中的最小值
	if (2 * u <= sz && heap[t] > heap[2*u]) t = 2 * u;
	if (2 * u + 1 <= sz && heap[t] > heap[2*u+1]) t = 2 * u + 1;
	//若找出的最小值不是本节点，则交换与最小节点的位置，并递归调用
	if (u != t) swap(heap[u], heap[t]), down(t);
}

//up同理
void up(int x) {
	//若为根节点，返回
	if (x <= 1) return;
	//每次与父节点相比，若比父节点小，则交换并递归父节点
	if (heap[x] < heap[x/2]) swap(heap[x], heap[x/2]), up(x/2);
}

返回目录