用C++的类做三种优先队列的实现

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了用C++的类做三种优先队列的实现相关的知识,希望对你有一定的参考价值。

学过数据结构的都知道优先队列这种东西,普通的队列是依据入队顺序,先入队的先出队,而优先队列则是依照键值,键值越大(或越小),就越先出队。

所以,优先队列基本支持push,pop,empty,size,top,这几种操作。最近在看C++prime,学了类之后觉得非常适合用来实现高级数据结构,于是就动手做了一下,花了一周,终于弄好了。以下的实现都默认最小堆。

二叉堆:

最常用,最简单的堆实现,因为二叉堆是完全二叉树,所有可以使用顺序存储结构,也就是说,数组来保存数据。二叉堆主要基于上滤与下滤操作来维持堆序(任意节点大与等于或小于等于他的子节点)而完全二叉树基本是平衡树,所以这些操作都是log2 N的时间复杂度,最后顺便一提,二叉堆可用于实现堆排序的算法。

技术分享
#include <iostream>
#include <vector>

class Bheap{//默认最小堆
public:
    typedef int Element;
    Bheap(const Element &Most);
    Bheap(Element *arr, int size, const Element &Most);
    ~Bheap();
    void push(const Element &e);
    void pop();
    bool empty();
    const Element& top();
    int size();
    void sort();
private:
    std::vector<Element> data;
    Element MostEle;
    void PercolateDown(int loca);
    void PercolateUp(int loca);
};

void Bheap::PercolateDown(int loca){
    int j, cur = data[loca];
    for (j = loca; j*2<data.size();){
        j *=2;
        if (j + 1 < data.size() && data[j + 1] < data[j]){
            j++;
        }
        if (data[j]<cur)
            data[j / 2] = data[j];
        else{
            j /= 2;
            break;
        }
    }
    data[j] = cur;
}
void Bheap::PercolateUp(int loca){
    int e = data[loca];
    while (1){
        if (e<data[loca /2]){
            data[loca] = data[loca / 2];
            loca /=2;
        }
        else{
            break;
        }
    }
    data[loca] = e;
}
Bheap::~Bheap()
{
}
Bheap::Bheap(Element *arr, int size, const Element &Most) :data(1, Most), MostEle(Most){
    for (int i = 0; i < size; i++){
        data.push_back(arr[i]);
    }
    for (int i = ((this->size()) /2); i > 0; i--){
        PercolateDown(i);
    }
}
Bheap::Bheap(const Element& Most) : data(1, Most), MostEle(Most)
{
}
void Bheap::push(const Element &e){
    data.push_back(e);
    PercolateUp(data.size() - 1);
}
void Bheap::pop(){
    data[1] = data[data.size() - 1];
    data.resize(data.size() - 1);
    PercolateDown(1);
}
bool Bheap::empty(){
    return data.size() <= 1;
}
const Bheap::Element& Bheap::top(){
    return data[1];
}
int Bheap::size(){
    return data.size() - 1;
}
void Bheap::sort(){
    for (int i = data.size() - 1; i > 0; i--){
        Element t = data[1];
        data[1] = data[i];
        data[i] = t;
        int j, cur = data[1];
        for (j = 1; j * 2<i;){
            j *=2;
            if (j + 1 < i && data[j + 1] < data[j]){
                j++;
            }
            if (data[j]<cur)
                data[j / 2] = data[j];
            else{
                j /= 2;
                break;
            }
        }
        data[j] = cur;
    }
}
View Code

左式堆:

左式堆是一颗二叉树,一般来说,我们更都喜欢平衡的树,平衡的树深度小,各种操作起来效率都更高,但是,左式堆却反其道而行之,最求不平衡的树,在实现avl树时,我们使用了左右子树的深度来作为树是否“比较平衡”的指标,而在左式堆中,我们也有类似的指标,npl(NULL path length),规定子节点数为0与1的节点的npl为0,空节点的npl为-1,任意节点的npl为他所有子节点中npl小的节点的npl+1。左式堆就是保持任意节点的左子树npl始终大于等于右子树的npl,这样的效果就是左子树始终比右子树深,虽然左子树深了,但是右子树就浅了啊,于是就可以大概保证push,pop操作为log2N,而且,不同于二叉堆,左式堆可以高效地支持合并操作,时间复杂度也是log2N左右,甚至,左式堆的所有操作都是基于合并操作。那么具体是怎么合并呢?因为是堆,于是保持着堆序,对于两个左式堆,我们比较根处的值,将小的堆的右子树与大的堆合并,并作为小的堆新的右子树,并且要更新npl,并根据npl决定是否交换子树,上面说过,左子树深了,右子树就浅了,于是就不用担心递归实现会爆栈来,因为,右子树非常浅(不大于log2N),深到递归实现会爆栈所需的数据是夸张的,所有可以放心地递归实现,但是我用了堆栈做了非递归实现(说了那么多,你居然没有用递归)。

技术分享
#include <iostream>
#include <vector>
#include <stack>

class Lheap
{
private:
    typedef int Element;
    struct LheapNode
    {
        Element e;
        LheapNode*left;
        LheapNode*right;
        int npl;
        LheapNode();
        ~LheapNode();
        void swapchild();
        void renewnpl();
        const LheapNode& operator = (const LheapNode&a);
    };
    LheapNode *node;
    int Size;
    void Merge1(Lheap &a);
public:
    Lheap();
    Lheap(const Element &E);
    Lheap(const LheapNode* a);
    Lheap(const Element *arr, int size);
    ~Lheap();
    void Merge(const Lheap &a);
    void pop();
    const Element &top();
    bool empty();
    int size()const;
    void push(const Element &E);
};

Lheap::LheapNode::LheapNode() :left(NULL), right(NULL), npl(0){}
Lheap::LheapNode::~LheapNode(){
    delete left;
    delete right;
}
void Lheap::LheapNode::renewnpl(){
    if (right){
        if (left){
            if (right->npl > left->npl){
                swapchild();
            }
            npl = right->npl + 1;
        }
        else{
            swapchild();
            npl = 0;
        }
    }
    else{
        npl = 0;
    }
}
void Lheap::LheapNode::swapchild(){
    LheapNode* t = left;
    left = right;
    right = t;
}
const Lheap::LheapNode& Lheap::LheapNode:: operator = (const LheapNode&a){
    if (a.left){
        if (left == NULL)
            left = new LheapNode;
        *left = *a.left;
    }
    if (a.right){
        if (right == NULL)
            right = new LheapNode;
        *right = *a.right;
    }
    npl = a.npl;
    e = a.e;
    return *this;
}

Lheap::Lheap() :node(NULL),Size(0)
{
}
Lheap::Lheap(const Element *arr, int size) : node(NULL), Size(0)
{
    node = new LheapNode;
    node->e = arr[0];
    for (int i = 1; i < size; i++)
        push(arr[i]);
}
Lheap::Lheap(const Element &E) : node(NULL), Size(0){
    node = new LheapNode;
    node->e = E;
}
Lheap::~Lheap()
{
    delete node;
}
bool Lheap::empty(){
    return node == NULL;
}
void Lheap::Merge1(Lheap &a){
    if (node == NULL){
        node = a.node;
        a.node = NULL;
        return;
    }
    std::stack<LheapNode*> s;
    s.push(node);
    LheapNode *b = a.node,*t;
    while (1){
        if (s.top()->e < b->e){
            if (s.top()->right){
                s.push(s.top()->right);
            }
            else{
                break;
            }
        }
        else{
            if (b->right){
                t = s.top();s.pop();
                s.push(b);
                b = t;
                s.push(s.top()->right);
            }
            else{
                t = s.top();s.pop();
                s.push(b);
                b = t;
                break;
            }
        }
    }
    while (!s.empty()){
        t = s.top(); s.pop();
        t->right = b;
        b = t;
        b->renewnpl();
    }
    node = b;
    a.node = NULL;
}
void Lheap::Merge(const Lheap &a){
    Lheap t;
    t.node = new LheapNode;
    *t.node = *a.node;//此处可根据需要选择是否要保留被并入的堆,不需要的话直接Merge1(a)就好了
    Merge1(t);
    Size += a.Size;
}
void Lheap::push(const Element &E){
    Lheap t(E);
    Merge1(t);
    Size++;
}
void Lheap::pop(){
    LheapNode *L = node->left,*R=node->right;
    node->left = node->right =NULL;
    delete node;
    node = L;
    if (R){
        Lheap t;
        t.node = R;
        Merge1(t);
    }
    Size--;
}
const Lheap::Element& Lheap::top(){
    return node->e;
}
int Lheap::size()const{
    return Size;
}
View Code

二项队列:

二项树,是一种特殊的树,他的特点是,我们把一棵有2^k个节点的树叫做K树,k树有k-1个子树,分别是0到k-1树,而二项队列就是按k树k值大小排列的队列(说好的队列呢),此外,二项队列有一个最大的特点,那就是每种树都只能存在一棵,如果存在两颗呢?那我们就要将他们合并为一棵树,比如说,如果有两颗0树,那么,我们就要将他们合并为一棵1树,怎么合并呢?想想,我们要构造的是堆,于是二项树就必须符合堆序,那我们就只要将根值大的作为小的的子树。说到这些,有没有觉得熟悉呢?我们看一下二进制的加法,

1001  8+0+0+1

0011  0+0+2+1

1100  8+4+0+0

明白了吧,二项队列的具体结构与插入的值无关,虽然他是个堆,但是他长得怎么样,和你给他什么值没什么关系,很奇怪对不对,但真的是这样,二项队列的结构取决于插入元素的个数。二项队列的特质,使他唯一对应于一个二进制数字,于是我们可以利用两个堆大小对应二进制数字的加法来控制流程。然后,我们对它的操作做分析,首先对合并操作进行分析,首先,我们知道,对于一个有N个元素的二项队列,长度不超过Log2N,因为,一个二项队列对应一个二进制整数。合并相当于一次加法运算,二项队列的合并就是对其队列中的二项树自小到大进行合并,插入下一位的操作(进位),那么复杂度是O(log2N),pop操作则是在队列中寻找根最小的二项树(花费Log2N),将其子树当成新队列,与除去了二项队列中该树,再将该队列与新队列合并,操作也是log2N,最后,就是push操作,二项队列最大的特点就是,虽然不能保证每次操作都是o(1)的时间复杂度,但是可以保证M次push操作,平均的时间复杂度是(MLog2N),为什么呢?我们来分析一下,对于push操作,相当于一个二项队列与对应1的二项队列相加。首先,先明白这样一件事,push操作花费的时间取决于二项队列结尾出现第一个0之前1的个数,比如说,00就只需要一次插入,01就需要一次插入,一次合并,011就需要一次插入,两次合并,假设一个二项队列在进行push操作前大小是k,通过M次插入操作变为k+M,该过程中,最低位为1的概率有1/2,于是有一半的数需要一次合并,而显然剩下的树不需要再合并,然后在这一半的数中,又有一半的数第二位为1,他们需要第二次合并,而另一半则不需要,依此类推,从k到k+M-1这M个数,所需合并次数为 M/2+M/4+M/8....+1,假设M为偶数(因为奇数结果也差不了多少),则为M(1+1/2+1/4+。。。+1/Log2M)=M-1,于是M次合并,平均需要O(M-1/M)=O(1)的时间复杂度。这种分析方式就是所谓的摊还分析了,有的数据结构,比如伸展树,比如斜堆,都有虽然不能保证每次操作是O(x),但是能保证M次操作为O(Mx)的特点,这种分析比较复杂,我到现在都不是很会,于是你会发现我在分析上面两种实现时草草带过,二叉堆是因为真的没有什么好说的,左式堆是因为我真的没有什么会说的。。。还是好好学习一下怎么进行摊还分析吧。另外,优先队列的实现中没有斜堆,因为他和左式堆差不多,区别就像AVL树与伸展树差不多,他不会通过npl这种东西来判断是否交换子树,他只是简单粗暴地交换每个合并的节点的子树。。。。最后吐槽一下,说好的队列,结果我使用deque来作为队列,但最后才发现,直接开个32大小的数组反而比较实在,毕竟可以存2^31大约10亿个数据了,对于我这种没见过世面的人来说已经妥妥的了。

技术分享
#include <iostream>
#include <vector>
const int Default_size=32;
class Bqueue
{
private:
    typedef int Element;
    struct BtreeNode
    {
        BtreeNode*child;
        BtreeNode*sibling;
        Element data;
        BtreeNode();
        ~BtreeNode();
    };
    typedef BtreeNode * Btree;
    std::vector<Btree> q;
    int Size;
public:
    Bqueue();
    Bqueue(int size);
    Bqueue(Element *arr, int size);
    ~Bqueue();
    void push(const  Element &e);
    void pop();
    Bqueue& Merge(Bqueue &Q);
    int size()const;
    const  Element& top()const;
    bool empty()const;
    Btree combineTree(Btree T1, Btree T2)const;
};

Bqueue::BtreeNode::BtreeNode():child(NULL), sibling(NULL){}
Bqueue::BtreeNode::~BtreeNode(){
    delete child;
    delete sibling;
}

Bqueue::Bqueue() :Size(0){
    q.resize(Default_size);
}
Bqueue::Bqueue(int size) : Size(size){
    q.resize(Default_size);
}
Bqueue::~Bqueue(){
    for (std::vector<Btree>::iterator p = q.begin(); p != q.end(); p++)
        delete *p;
}
Bqueue& Bqueue::Merge(Bqueue &Q){
    Btree T1, T2,Carry=NULL;
    Size += Q.size();
    for (int i = 0, j = 1;j<=Size; i++, j *= 2){
        T1 = q[i]; T2 = Q.q[i];
        switch (!!T1+2*!!T2+4*!!Carry)
        {
        case 0:
        case 1:
            break;
        case 2:
            q[i] = T2;
            Q.q[i] = NULL;
            break;
        case 3:
            Carry = combineTree(T1, T2);
            q[i] = Q.q[i] = NULL;
            break;
        case 4:
            q[i] = Carry;
            Carry = NULL; 
            break;
        case 5:
            Carry = combineTree(T1, Carry);
            q[i] = NULL;
            break;
        case 6:
            Carry = combineTree(Carry, T2);
            Q.q[i] = NULL;
            break;
        case 7:
            q[i] = Carry;
            Carry = combineTree(T1, T2);
            Q.q[i] = NULL;
            break;
        default:
            break;
        }
    }
    return *this;
}
int Bqueue::size()const{
    return Size;
}
void Bqueue::push(const Element &e){
    Btree T = new BtreeNode,T1;
    T->data = e;
    Size += 1;
    for (int i = 0,j=1; j <= Size&&T; i++,j*=2){
        T1 = q[i];
        switch (!!T*2+!!T1){
        case 2:
            q[i] = T;
            T = NULL;
                break;
        case 3:
            q[i] = NULL;
            T=combineTree(T, T1);
                break;
        default:
            break;
        }
    }
}
void Bqueue::pop(){
        int i, j, Mintree=0;
        Btree DeletedTree, Oldroot;
        while (q[Mintree]==NULL)
        {
            Mintree++;
        }
        for (i = Mintree+1, j = 1<<i; j <= Size; j *= 2,i++)
            if (q[i])
            Mintree = q[Mintree]->data<q[i]->data?Mintree:i;
        Oldroot = DeletedTree = q[Mintree];
        DeletedTree = DeletedTree->child;
        Oldroot->child = NULL;
        delete Oldroot;
        Bqueue DeleteQueue((1 << Mintree) - 1);
        for (i = Mintree - 1; i >= 0; i--){
            DeleteQueue.q[i] = DeletedTree;
            DeletedTree = DeletedTree->sibling;
            DeleteQueue.q[i]->sibling = NULL;
        }
        q[Mintree] = NULL;
        Size -= DeleteQueue.size() + 1;
        Merge(DeleteQueue);
}
bool Bqueue::empty()const{
    return Size == 0;
}
const Bqueue:: Element& Bqueue::top()const{
    int res=0;
    while (q[res] == NULL)
    {
        res++;
    }
    for (int i = res + 1, j = 1 << i; j <= Size; j *= 2, i++)
        if (q[i])
            res = q[res]->data<q[i]->data ? res : i;
    return q[res]->data;
}
Bqueue::Btree Bqueue::combineTree(Btree T1,Btree T2)const{
        if (T1->data > T2->data)
            return  combineTree(T2, T1);
        T2->sibling = T1->child;
        T1->child = T2;
        return T1;
    }
View Code

最后,我们实际看一下各种实现的效果,我们用合并果子这道经典贪心算法来测试一下。

二叉堆:

  • 测试点1 Accepted / 1ms / 12396kB
  • 测试点2 Accepted / 2ms / 12396kB
  • 测试点3 Accepted / 2ms / 12396kB
  • 测试点4 Accepted / 19ms / 12396kB
  • 测试点5 Accepted / 18ms / 12396kB
  • 测试点6 Accepted / 51ms / 12396kB
  • 测试点7 Accepted / 50ms / 12396kB
  • 测试点8 Accepted / 45ms / 12396kB
  • 测试点9 Accepted / 53ms / 12396kB
  • 测试点10 Accepted / 50ms / 12396kB

STL 的优先队列《vector》

  

  • 测试点1 Accepted / 5ms / 12400kB
  • 测试点2 Accepted / 8ms / 12400kB
  • 测试点3 Accepted / 14ms / 12400kB
  • 测试点4 Accepted / 36ms / 12400kB
  • 测试点5 Accepted / 45ms / 12400kB
  • 测试点6 Accepted / 85ms / 12400kB
  • 测试点7 Accepted / 84ms / 12400kB
  • 测试点8 Accepted / 80ms / 12400kB
  • 测试点9 Accepted / 85ms / 12400kB
  • 测试点10 Accepted / 87ms / 12400kB

二项队列:

  

  • 测试点1 Accepted / 3ms / 12396kB
  • 测试点2 Accepted / 17ms / 12396kB
  • 测试点3 Accepted / 20ms / 12396kB
  • 测试点4 Accepted / 77ms / 12396kB
  • 测试点5 Accepted / 101ms / 12528kB
  • 测试点6 Accepted / 202ms / 12660kB
  • 测试点7 Accepted / 196ms / 12660kB
  • 测试点8 Accepted / 195ms / 12660kB
  • 测试点9 Accepted / 211ms / 12660kB
  • 测试点10 Accepted / 190ms / 12660kB

左式堆:
  

  • 测试点1 Accepted / 2ms / 12400kB
  • 测试点2 Accepted / 29ms / 12400kB
  • 测试点3 Accepted / 30ms / 12400kB
  • 测试点4 Accepted / 143ms / 12528kB
  • 测试点5 Accepted / 164ms / 12528kB
  • 测试点6 Accepted / 391ms / 12784kB
  • 测试点7 Accepted / 379ms / 12784kB
  • 测试点8 Accepted / 381ms / 12784kB
  • 测试点9 Accepted / 376ms / 12784kB
  • 测试点10 Accepted / 357ms / 12784kB

可以看出,简单除暴的二叉堆效率是最高的,左式堆最不给力。。。

 

以上是关于用C++的类做三种优先队列的实现的主要内容,如果未能解决你的问题,请参考以下文章

C++ 我在设置我创建的类的优先级队列时遇到问题

基于最大堆实现最大优先队列代码

如何在 C++ 中正确使用优先级队列

c++ 将指针推入指针优先级队列会导致立即 valgrind 错误

C++队列和优先权队列的使用---应用:带时限作业排序

(c++)迷宫自动寻路-队列-广度优先算法-附带寻路打印动画