文章目录
二叉堆
前言
数据结构与算法是每一个CS人的必经之路
路漫漫其修远兮,今天总结的是关于数据结构的二叉堆
一、什么是二叉堆
二叉堆是一种支持插入、删除、查询最值的数据结构
实际上 二叉堆是一种满足堆结构的完全二叉树
此二叉树的每个节点带有权值,每个父节点下有两个子节点(二叉树性质)
其满足每个父节点的权值 ≥ 或者 ≤ 其任意子节点的权值,则称其具有“大根堆性质”或”小根堆性质“
对于这种数据结构,我们通过上浮,下沉,插入等操作始终维护一个具有以上性质的树结构,所以对于二叉堆,我们查询数列最值的时间复杂度是O(1)的,而维护一个长度为N,满足二叉堆性质的数组,其增加,删除操作均为O(logN)
二、二叉堆的建立、维护、查询
1.堆的维护
代码如下:
根据完全二叉树的性质,我们很轻易的知道,以1为根节点的二叉树,每一个子节点的节点下标皆为2*K/ 2*K+1
int heap[N];//堆数组
int tot;//堆的大小
void swim(int n){//上浮
for(int i=n;i>1&&heap[i]>heap[i/2];i/=2)
swap(heap[i],heap[i/2]);
//直至找到当前数值能成为父节点的位置 不断交换即可
}
int findson(int n){//查找交换节点
return n*2 +(2*n+1<=tot && heap[n*2+1]>heap[n*2]);
//判断当前节点的左子节点或右子节点
//在存在右子节点的情况,如果右子节点大,就为1 那么查找的交换节点就是右子节点 反之亦然
}
void sink(int n){//下沉
//n记为操作节点
int i,t;
for(i=n,t=findson(i);t<=tot && heap[t]>heap[i]; i=t,t=findson(i));
//从当前节点出发查找,判断当前节点的子节点,找到子节点中最大的那个
//如果当前节点小于此子节点,交换(下沉),不断操作直至操作节点下沉至合适位置
swap(heap[i],heap[t]);
}
void build(int A[],int n){//从0-index O(n)地建立二叉堆
memcpy(heap+1,A,sizeof(int) *n);
tot=n;
for(int i=n/2;i>0;--i)
sink(i);//下沉即可
}
2.数据的插入与查询
代码如下(示例):
void insert(int x){//插入
heap[++tot]=x;
swim(x);
//插入堆尾,再执行上浮操作即可
}
void pop(){
swap(heap[1],heap[tot--]);
sink(1);
//交换堆顶与堆尾,再下沉即可
}
int top(){//查询
return heap[1];
}
3.STL
c++的STL库中已经封装好了具有大根堆和小根堆性质的优先队列
priority_queue<int, vector<int>, less<int>>s;
//less表示按照递减(从大到小)的顺序插入元素
priority_queue<int, vector<int>, greater<int>>s;
//greater表示按照递增(从小到大)的顺序插入元素
总结
以上就是今天要讲的内容,二叉堆是一个很好的维护队列最值的数据结构,在数据庞大是是维护区间最值的不二之选。