JS实现堆:
JS中没有封装好的队列、栈和堆。队列和栈可以用数组模拟,但是堆需要自己封装数据结构。或者封装一个构造函数,每次建堆都是一个实例化的过程。
在JAVA和C++中也都提供了优先队列这个数据结构,这个数据结构的底层实现也是用堆完成的。
下面我们用JS模拟一个堆以实现优先队列。
堆的几个特性:
- 堆本身是一棵满二叉树,这里我们实现一个小根堆
- 左孩子索引 = (父元素索引+1)*2-1
- 右孩子索引 = (父元素索引+1)*2
- 插入元素放在树的最后(树是用数组实现的,也就是push到数组的最后面),然后比较这个节点和父节点的大小,小于父节点就交换父子节点。直到父节点为根节点。
- 删除元素删除的是根节点,根节点和树的最后一个节点(last)交换位置,然后存储树的数组长度-1,这样树中就没有被删除的根节点。被交换的last节点和子元素中的最小值比较,如果last节点大于最小值,交换两节点,直到不存在子元素。完成删除操作。
function Heap()
{
this.data = [];
}
Heap.prototype.print = function () {
console.log("Heap: " + this.data);
}
Heap.prototype.build = function(data){
// 初始化
this.data = [];
if (!data instanceof Array)
return false;
// 入堆
for (var i = 0; i < data.length; ++i) {
this.insert(data[i]);
}
return true;
}
Heap.prototype.insert = function( nValue ){
if (!this.data instanceof Array) {
this.data = [];
}
this.data.push(nValue);
// 更新新节点
var nIndex = this.data.length-1;
var nFatherIndex = Math.floor((nIndex-1)/2);
while (nFatherIndex > 0){
if (this.data[nIndex] < this.data[nFatherIndex]) {
var temp = this.data[nIndex];
this.data[nIndex] = this.data[nFatherIndex];
this.data[nFatherIndex] = temp;
}
nIndex = nFatherIndex;
nFatherIndex = Math.floor((nIndex-1)/2);
}
}
Heap.prototype.delete = function( ){
if (!this.data instanceof Array) {
return null;
}
var nIndex = 0;
var nValue = this.data[nIndex];
var nMaxIndex = this.data.length-1;
// 更新新节点
var nLeaf = this.data.pop();
this.data[nIndex] = nLeaf;
while (nIndex < nMaxIndex ){
var nLeftIndex = 2 * (nIndex+1) - 1;
var nRightIndex = 2 * (nIndex+1);
// 找最小的一个子节点(nLeftIndex < nRightIndex)
var nSelectIndex = nLeftIndex;
if (nRightIndex < nMaxIndex) {
nSelectIndex = (this.data[nLeftIndex] > this.data[nRightIndex]) ? nRightIndex : nLeftIndex;
}
if (nSelectIndex < nMaxIndex && this.data[nIndex] > this.data[nSelectIndex] ){
var temp = this.data[nIndex];
this.data[nIndex] = this.data[nSelectIndex];
this.data[nSelectIndex] = temp;
}
nIndex = nSelectIndex;
}
return nValue;
}
测试用例:
var heap = new Heap();
heap.build([1,2,5,7,9,11,13,15]);
heap.print();
console.log('----------')
// insert
heap.insert(4);
heap.print();
console.log('----------')
// delete
heap.delete();
heap.print();