堆-1

2017-08-17  本文已影响0人  SylviaShen

堆(优先级队列,Priority Queue)是一个完全二叉树另加上一个条件:父节点的值总比两个子节点的值大(或者小,这里用大)。由于是完全二叉树,它和它的层次遍历是完全一一对应的,就可以在物理上直接用线性表保存。希望看成树的时候,数一项在一层,数两项在下一层,再四项八项地数,一层一层堆下去,就可以唯一地还原树的结构。在列表里,下标为i的元素的左、右孩子(若有)分别是2 * i + 12 * i + 2, 父亲(0没有)下标是(i - 1) / 2。脑子里想着树,物理上用线性表实现就可以了。

堆维护的是一种“偏序性”,并不需要完全按照次序排序。这一点使它在很多操作上能做到比有序数组或者链表更快。

上滤percolate up,下滤percolate down

堆怎么使用?

下面用到的宏定义,和上滤操作:

#define parent(i) (((i) - 1) / 2)
#define lc(i) (2 * (i) + 1)
#define rc(i) (2 * (i) + 2)

void percolateUp(int heap[], int idx, int len){
    int temp = heap[idx];
    while(idx > 0 && temp > heap[parent(idx)]){
        heap[idx] = heap[parent(idx)];
        idx = parent(idx);    
    }
    heap[idx] = temp;
}

下滤操作,用idxMax表示自己以及左右孩子中最大的那个的秩:

void percolateDown(int heap[], int idx, int len){
    int temp = heap[idx], idxMax;
    while(lc(idx) < len){
        if(rc(idx) >= len){
            idxMax = (temp < heap[lc(idx)]? lc(idx): idx);
        }else{
            if(heap[lc(idx)] < heap[rc(idx)]){
                idxMax = (temp < heap[rc(idx)]? rc(idx): idx);
            }else{
                idxMax = (temp < heap[lc(idx)]? lc(idx): idx);
            }
        }
        if(idxMax == idx) break;
        heap[idx] = heap[idxMax];
        idx = idxMax;
    }
    heap[idx] = temp;
}

建堆

817.png

这个复杂度好在哪里了呢?最坏情况下的操作次数是(1 * 2 ^ (h-1)) + (2 * 2 ^ (h-2) ) + ... + (h * 1)。和刚才刚好反了,节点最多的最后一层,操作最少;节点少的堆顶部分,操作次数多。或者说,前者上滤的操作次数是深度,后者的操作次数是高度。这个式子放缩一下,可以得到复杂度是n。爽。

代码:

void makeHeap_Force(int heap[], int len){ //自上而下的上滤
    for(int i = 0; i < len; i ++){
        percolateUp(heap, i, len);
    }
}

void makeHeap_Floyd(int heap[], int len){ //自下而上的下滤
    //下滤的次序是:从最后一个非叶子节点开始,依次向前
    for(int i = (len - 2) / 2; i >= 0; i --){
        percolateDown(heap, i, len);
    }
}

堆排序

这个思路可以是基于选择法排序来的。选择法排序的时候,可以不断地在前面x[0:i)里面选出最大值,放进x[i], 然后i --。(虽然我的习惯是选最小值往前放,而不是选最大值往后放, 都一样的)

这里呢,就把遍历前缀然后选出最大值,改成了从堆里拿出最大值。把数组建成堆之后,x[0]就是最大元素,准备放进x[i],正好,为了维护这个堆,要把x[i]补到堆顶,然后下滤。它俩一交换,下滤,就可以了。

循环每次耗时log(n),因此复杂度n * log(n)。好像和快排差不多。然而堆排序可以做到就地排序,只需要常数的额外空间。而且据说在实际运行中效率是比计算的要好的。反正我只知道很漂亮的样子……

void HeapSort(int heap[], int len){
    makeHeap_Floyd(heap, len);
    int temp;
    for(int i = len - 1; i > 0; i --){
        temp = heap[i];
        heap[i] = heap[0];
        heap[0] = temp;
        percolateDown(heap, 0, i);
    }
}

怎么又是这么多内容啊天,写得累死我了,下次明白的地方我一定简略点。

上一篇 下一篇

猜你喜欢

热点阅读