JS实现堆

JS实现堆:

JS中没有封装好的队列、栈和堆。队列和栈可以用数组模拟,但是堆需要自己封装数据结构。或者封装一个构造函数,每次建堆都是一个实例化的过程。
在JAVA和C++中也都提供了优先队列这个数据结构,这个数据结构的底层实现也是用堆完成的。
下面我们用JS模拟一个堆以实现优先队列。

堆的几个特性:

  1. 堆本身是一棵满二叉树,这里我们实现一个小根堆
  2. 左孩子索引 = (父元素索引+1)*2-1
  3. 右孩子索引 = (父元素索引+1)*2
  4. 插入元素放在树的最后(树是用数组实现的,也就是push到数组的最后面),然后比较这个节点和父节点的大小,小于父节点就交换父子节点。直到父节点为根节点。
  5. 删除元素删除的是根节点,根节点和树的最后一个节点(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();
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值