1 堆的概念
如果有一个关键码的集合K = {k0,k1, k2,…,kn-1},把它的所有元素按完全二叉树的顺序存储方式存储 在一个一维数组中,并满足:Ki <= K2i+1 且 Ki<= K2i+2 (Ki >= K2i+1 且 Ki >= K2i+2) i = 0,1,2…,则称为 小堆(或大堆)。将根节点最大的堆叫做最大堆或大根堆,根节点最小的堆叫做最小堆或小根堆。
v堆的性质:
- 堆中某个节点的值总是不大于或不小于其父节点的值;
- 堆总是一棵完全二叉树。
2 堆的存储方式
从堆的概念可知,堆是一棵完全二叉树,因此可以层序的规则采用顺序的方式来高效存储,
注意:对于非完全二叉树,则不适合使用顺序方式进行存储,因为为了能够还原二叉树,空间中必须要存储空节点,就会导致空间利用率比较低。
将元素存储到数组中后,可以根据二叉树章节的性质5对树进行还原。假设i为节点在数组中的下标,则有:
- 如果i为0,则i表示的节点为根节点,否则i节点的双亲节点为 (i - 1)/2
- 如果2 * i + 1 小于节点个数,则节点i的左孩子下标为2 * i + 1,否则没有左孩子
- 如果2 * i + 2 小于节点个数,则节点i的右孩子下标为2 * i + 2,否则没有右孩子
3 堆的插入与删除
4.1 堆的插入
堆的插入总共需要两个步骤:
- 先将元素放入到底层空间中(注意:空间不够时需要扩容)
- 将最后新插入的节点向上调整,直到满足堆的性质
4.2 堆的删除
注意:堆的删除一定删除的是堆顶元素。具体如下:
- 将堆顶元素对堆中最后一个元素交换
- 将堆中有效数据个数减少一个
- 对堆顶元素进行向下调整
4 代码实现
public class TestHeap {
public int[] elem;
public int usedSize;
public TestHeap() {
this.elem = new int[10];
}
/**
* 时间复杂度:O(logn)
* @param root 每棵子树的开始位置
* @param len 结束位置
*/
public void adjustDown(int root,int len) {
int parent = root;
int child = 2*parent+1;
while (child < len) {
//0、判断是否有左右孩子 有的话 找到最大值 C下标表示最大值下标
if(child+1 < len && this.elem[child] < this.elem[child+1]) {
child++;
}
//代码指向到这里,c表示最大值下标
if(this.elem[child] > this.elem[parent]) {
//交换
int tmp = this.elem[child];
this.elem[child] = this.elem[parent];
this.elem[parent] = tmp;
parent = child;
child = 2*parent+1;
}else {
break;
}
}
}
//O(n)
public void createHeap(int[] array) {
for (int i = 0; i < array.length; i++) {
this.elem[i] = array[i];
this.usedSize++;
}
//i:每棵子树的根节点下标
for (int i = (this.usedSize-1-1)/2; i >= 0 ; i--) {
adjustDown(i,this.usedSize);
}
}
/**
* 插入数据:
* @param val
*/
public void push(int val) {
//0、堆是否是满的--》扩容
if(isFull()) {
this.elem = Arrays.copyOf(this.elem,2*this.elem.length);
}
//1、放到数组的最后一个位置
this.elem[this.usedSize] = val;
this.usedSize++;//11
//2、进行调整
adjustUp(this.usedSize-1);
}
public void adjustUp(int child) {
int parent = (child-1)/2;
while (child > 0) {
if(this.elem[child] > this.elem[parent]) {
int tmp = this.elem[child];
this.elem[child] = this.elem[parent];
this.elem[parent] = tmp;
child = parent;
parent = (child-1)/2;
}else {
break;
}
}
}
public boolean isFull() {
return this.usedSize == this.elem.length;
}
public void pop() {
//0、是否是空的
if(isEmpty()) {
return;
}
//1、最后一个元素和堆顶元素交换
int tmp = this.elem[0];
this.elem[0] = this.elem[this.usedSize-1];
this.elem[this.usedSize-1] = tmp;
this.usedSize--;//10 9
//2、调整0号下标这棵树
adjustDown(0,this.usedSize);
}
public boolean isEmpty() {
return this.usedSize == 0;
}
public int peek() {
if(isEmpty()) {
return -1;
}
return this.elem[0];
}
/**
* 时间复杂度:O(n*logn)
* 空间复杂度:O(1)
*/
public void heapSort() {
int end = this.usedSize-1;
while (end > 0) {
int tmp = this.elem[0];
this.elem[0] = this.elem[end];
this.elem[end] = tmp;
adjustDown(0,end);
end--;
}//O(n*logn)
}
public void show() {
for (int i = 0; i < this.usedSize; i++) {
System.out.print(this.elem[i]+" ");
}
System.out.println();
}
}