斜堆(二)之 C++的实现

 

概要

上一章介绍了斜堆的基本概念，并通过C语言实现了斜堆。本章是斜堆的C++实现。

目录
1. 斜堆的介绍
2. 斜堆的基本操作
3. 斜堆的C++实现(完整源码)
4. 斜堆的C++测试程序

转载请注明出处：http://www.cnblogs.com/skywang12345/p/3638524.html

---

更多内容：数据结构与算法系列 目录

 

斜堆的介绍

斜堆(Skew heap)也叫自适应堆(self-adjusting heap)，它是左倾堆的一个变种。和左倾堆一样，它通常也用于实现优先队列；作为一种自适应的左倾堆，它的合并操作的时间复杂度也是O(lg n)。
它与左倾堆的差别是：
(01) 斜堆的节点没有"零距离"这个属性，而左倾堆则有。
(02) 斜堆的合并操作和左倾堆的合并操作算法不同。

斜堆的合并操作
(01) 如果一个空斜堆与一个非空斜堆合并，返回非空斜堆。
(02) 如果两个斜堆都非空，那么比较两个根节点，取较小堆的根节点为新的根节点。将"较小堆的根节点的右孩子"和"较大堆"进行合并。
(03) 合并后，交换新堆根节点的左孩子和右孩子。
       第(03)步是斜堆和左倾堆的合并操作差别的关键所在，如果是左倾堆，则合并后要比较左右孩子的零距离大小，若右孩子的零距离 > 左孩子的零距离，则交换左右孩子；最后，在设置根的零距离。

 

斜堆的基本操作

1. 基本定义

template <class T>
class SkewNode{
    public:
        T key;                // 关键字(键值)
        SkewNode *left;        // 左孩子
        SkewNode *right;    // 右孩子

        SkewNode(T value, SkewNode *l, SkewNode *r):
            key(value), left(l),right(r) {}
};

SkewNode是斜堆对应的节点类。

template <class T>
class SkewHeap {
    private:
        SkewNode<T> *mRoot;    // 根结点

    public:
        SkewHeap();
        ~SkewHeap();

        // 前序遍历"斜堆"
        void preOrder();
        // 中序遍历"斜堆"
        void inOrder();
        // 后序遍历"斜堆"
        void postOrder();

         // 将other的斜堆合并到this中。
        void merge(SkewHeap<T>* other);
        // 将结点(key为节点键值)插入到斜堆中
        void insert(T key);
        // 删除结点(key为节点键值)
        void remove();

        // 销毁斜堆
        void destroy();

        // 打印斜堆
        void print();
    private:

        // 前序遍历"斜堆"
        void preOrder(SkewNode<T>* heap) const;
        // 中序遍历"斜堆"
        void inOrder(SkewNode<T>* heap) const;
        // 后序遍历"斜堆"
        void postOrder(SkewNode<T>* heap) const;

        // 交换节点x和节点y
        void swapNode(SkewNode<T> *&x, SkewNode<T> *&y);
        // 合并"斜堆x"和"斜堆y"
        SkewNode<T>* merge(SkewNode<T>* &x, SkewNode<T>* &y);

        // 销毁斜堆
        void destroy(SkewNode<T>* &heap);

        // 打印斜堆
        void print(SkewNode<T>* heap, T key, int direction);
};

SkewHeap是斜堆类，它包含了斜堆的根节点，以及斜堆的操作。

 

2. 合并

/*
 * 合并"斜堆x"和"斜堆y"
 */
template <class T>
SkewNode<T>* SkewHeap<T>::merge(SkewNode<T>* &x, SkewNode<T>* &y)
{
    if(x == NULL)
        return y;
    if(y == NULL)
        return x;

    // 合并x和y时，将x作为合并后的树的根；
    // 这里的操作是保证: x的key < y的key
    if(x->key > y->key)
        swapNode(x, y);

    // 将x的右孩子和y合并，
    // 合并后直接交换x的左右孩子，而不需要像左倾堆一样考虑它们的npl。
    SkewNode<T> *tmp = merge(x->right, y);
    x->right = x->left;
    x->left  = tmp;

    return x;
}

/*
 * 将other的斜堆合并到this中。
 */
template <class T>
void SkewHeap<T>::merge(SkewHeap<T>* other)
{
    mRoot = merge(mRoot, other->mRoot);
}

merge(x, y)是内部接口，作用是合并x和y这两个斜堆，并返回得到的新堆的根节点。
merge(other)是外部接口，作用是将other合并到当前堆中。

3. 添加

/* 
 * 新建键值为key的结点并将其插入到斜堆中
 *
 * 参数说明：
 *     heap 斜堆的根结点
 *     key 插入的结点的键值
 * 返回值：
 *     根节点
 */
template <class T>
void SkewHeap<T>::insert(T key)
{
    SkewNode<T> *node;    // 新建结点

    // 新建节点
    node = new SkewNode<T>(key, NULL, NULL);
    if (node==NULL)
    {
        cout << "ERROR: create node failed!" << endl;
        return ;
    }

    mRoot = merge(mRoot, node);
}

insert(key)的作用是新建键值为key的节点，并将其加入到当前斜堆中。

 

4. 删除

/* 
 * 删除结点
 */
template <class T>
void SkewHeap<T>::remove()
{
    if (mRoot == NULL)
        return NULL;

    SkewNode<T> *l = mRoot->left;
    SkewNode<T> *r = mRoot->right;

    // 删除根节点
    delete mRoot;
    // 左右子树合并后的新树
    mRoot = merge(l, r); 
}

remove()的作用是删除斜堆的最小节点。

 

注意：关于斜堆的"前序遍历"、"中序遍历"、"后序遍历"、"打印"、"销毁"等接口就不再单独介绍了。后文的源码中有给出它们的实现代码，Please RTFSC(Read The Fucking Source Code)！

 

斜堆的C++实现(完整源码)

斜堆的实现文件(SkewHeap.h)

/**
 * C++: 斜堆
 *
 * @author skywang
 * @date 2014/03/31
 */

#ifndef _SKEW_HEAP_HPP_
#define _SKEW_HEAP_HPP_

#include <iomanip>
#include <iostream>
using namespace std;

template <class T>
class SkewNode{
    public:
        T key;                // 关键字(键值)
        SkewNode *left;        // 左孩子
        SkewNode *right;    // 右孩子

        SkewNode(T value, SkewNode *l, SkewNode *r):
            key(value), left(l),right(r) {}
};

template <class T>
class SkewHeap {
    private:
        SkewNode<T> *mRoot;    // 根结点

    public:
        SkewHeap();
        ~SkewHeap();

        // 前序遍历"斜堆"
        void preOrder();
        // 中序遍历"斜堆"
        void inOrder();
        // 后序遍历"斜堆"
        void postOrder();

         // 将other的斜堆合并到this中。
        void merge(SkewHeap<T>* other);
        // 将结点(key为节点键值)插入到斜堆中
        void insert(T key);
        // 删除结点(key为节点键值)
        void remove();

        // 销毁斜堆
        void destroy();

        // 打印斜堆
        void print();
    private:

        // 前序遍历"斜堆"
        void preOrder(SkewNode<T>* heap) const;
        // 中序遍历"斜堆"
        void inOrder(SkewNode<T>* heap) const;
        // 后序遍历"斜堆"
        void postOrder(SkewNode<T>* heap) const;

        // 交换节点x和节点y
        void swapNode(SkewNode<T> *&x, SkewNode<T> *&y);
        // 合并"斜堆x"和"斜堆y"
        SkewNode<T>* merge(SkewNode<T>* &x, SkewNode<T>* &y);

        // 销毁斜堆
        void destroy(SkewNode<T>* &heap);

        // 打印斜堆
        void print(SkewNode<T>* heap, T key, int direction);
};

/* 
 * 构造函数
 */
template <class T>
SkewHeap<T>::SkewHeap():mRoot(NULL)
{
}

/* 
 * 析构函数
 */
template <class T>
SkewHeap<T>::~SkewHeap() 
{
    destroy(mRoot);
}

/*
 * 前序遍历"斜堆"
 */
template <class T>
void SkewHeap<T>::preOrder(SkewNode<T>* heap) const
{
    if(heap != NULL)
    {
        cout<< heap->key << " " ;
        preOrder(heap->left);
        preOrder(heap->right);
    }
}

template <class T>
void SkewHeap<T>::preOrder() 
{
    preOrder(mRoot);
}

/*
 * 中序遍历"斜堆"
 */
template <class T>
void SkewHeap<T>::inOrder(SkewNode<T>* heap) const
{
    if(heap != NULL)
    {
        inOrder(heap->left);
        cout<< heap->key << " " ;
        inOrder(heap->right);
    }
}

template <class T>
void SkewHeap<T>::inOrder() 
{
    inOrder(mRoot);
}

/*
 * 后序遍历"斜堆"
 */
template <class T>
void SkewHeap<T>::postOrder(SkewNode<T>* heap) const
{
    if(heap != NULL)
    {
        postOrder(heap->left);
        postOrder(heap->right);
        cout<< heap->key << " " ;
    }
}

template <class T>
void SkewHeap<T>::postOrder() 
{
    postOrder(mRoot);
}

/*
 * 交换两个节点的内容
 */
template <class T>
void SkewHeap<T>::swapNode(SkewNode<T> *&x, SkewNode<T> *&y)
{
    SkewNode<T> *tmp = x;
    x = y;
    y = tmp;
}


/*
 * 合并"斜堆x"和"斜堆y"
 */
template <class T>
SkewNode<T>* SkewHeap<T>::merge(SkewNode<T>* &x, SkewNode<T>* &y)
{
    if(x == NULL)
        return y;
    if(y == NULL)
        return x;

    // 合并x和y时，将x作为合并后的树的根；
    // 这里的操作是保证: x的key < y的key
    if(x->key > y->key)
        swapNode(x, y);

    // 将x的右孩子和y合并，
    // 合并后直接交换x的左右孩子，而不需要像左倾堆一样考虑它们的npl。
    SkewNode<T> *tmp = merge(x->right, y);
    x->right = x->left;
    x->left  = tmp;

    return x;
}

/*
 * 将other的斜堆合并到this中。
 */
template <class T>
void SkewHeap<T>::merge(SkewHeap<T>* other)
{
    mRoot = merge(mRoot, other->mRoot);
}

/* 
 * 新建键值为key的结点并将其插入到斜堆中
 *
 * 参数说明：
 *     heap 斜堆的根结点
 *     key 插入的结点的键值
 * 返回值：
 *     根节点
 */
template <class T>
void SkewHeap<T>::insert(T key)
{
    SkewNode<T> *node;    // 新建结点

    // 新建节点
    node = new SkewNode<T>(key, NULL, NULL);
    if (node==NULL)
    {
        cout << "ERROR: create node failed!" << endl;
        return ;
    }

    mRoot = merge(mRoot, node);
}

/* 
 * 删除结点
 */
template <class T>
void SkewHeap<T>::remove()
{
    if (mRoot == NULL)
        return NULL;

    SkewNode<T> *l = mRoot->left;
    SkewNode<T> *r = mRoot->right;

    // 删除根节点
    delete mRoot;
    // 左右子树合并后的新树
    mRoot = merge(l, r); 
}

/* 
 * 销毁斜堆
 */
template <class T>
void SkewHeap<T>::destroy(SkewNode<T>* &heap)
{
    if (heap==NULL)
        return ;

    if (heap->left != NULL)
        destroy(heap->left);
    if (heap->right != NULL)
        destroy(heap->right);

    delete heap;
}

template <class T>
void SkewHeap<T>::destroy()
{
    destroy(mRoot);
}

/*
 * 打印"斜堆"
 *
 * key        -- 节点的键值 
 * direction  --  0，表示该节点是根节点;
 *               -1，表示该节点是它的父结点的左孩子;
 *                1，表示该节点是它的父结点的右孩子。
 */
template <class T>
void SkewHeap<T>::print(SkewNode<T>* heap, T key, int direction)
{
    if(heap != NULL)
    {
        if(direction==0)    // heap是根节点
            cout << setw(2) << heap->key << " is root" << endl;
        else                // heap是分支节点
            cout << setw(2) << heap->key << " is " << setw(2) << key << "'s "  << setw(12) << (direction==1?"right child" : "left child") << endl;

        print(heap->left, heap->key, -1);
        print(heap->right,heap->key,  1);
    }
}

template <class T>
void SkewHeap<T>::print()
{
    if (mRoot != NULL)
        print(mRoot, mRoot->key, 0);
}
#endif

斜堆的测试程序(SkewHeapTest.cpp)

/**
 * C 语言: 斜堆
 *
 * @author skywang
 * @date 2014/03/31
 */

#include <iostream>
#include "SkewHeap.h"
using namespace std;

int main()
{
    int i;
    int a[]= {10,40,24,30,36,20,12,16};
    int b[]= {17,13,11,15,19,21,23};
    int alen=sizeof(a)/sizeof(a[0]);
    int blen=sizeof(b)/sizeof(b[0]);
    SkewHeap<int>* ha=new SkewHeap<int>();
    SkewHeap<int>* hb=new SkewHeap<int>();

    cout << "== 斜堆(ha)中依次添加: ";
    for(i=0; i<alen; i++)
    {
        cout << a[i] <<" ";
        ha->insert(a[i]);
    }
    cout << "\n== 斜堆(ha)的详细信息: " << endl;
    ha->print();


    cout << "\n== 斜堆(hb)中依次添加: ";
    for(i=0; i<blen; i++)
    {
        cout << b[i] <<" ";
        hb->insert(b[i]);
    }
    cout << "\n== 斜堆(hb)的详细信息: " << endl;
    hb->print();


    // 将"斜堆hb"合并到"斜堆ha"中。
    ha->merge(hb);
    cout << "\n== 合并ha和hb后的详细信息: " << endl;
    ha->print();


    // 销毁
    ha->destroy();

    return 0;
}

 

斜堆的C++测试程序

斜堆的测试程序已经包含在它的实现文件(SkewHeapTest.cpp)中了，这里仅给出它的运行结果：

== 斜堆(ha)中依次添加: 10 40 24 30 36 20 12 16 
== 斜堆(ha)的详细信息: 
10 is root
16 is 10's   left child
20 is 16's   left child
30 is 20's   left child
40 is 30's   left child
12 is 10's  right child
24 is 12's   left child
36 is 24's   left child

== 斜堆(hb)中依次添加: 17 13 11 15 19 21 23 
== 斜堆(hb)的详细信息: 
11 is root
13 is 11's   left child
17 is 13's   left child
23 is 17's   left child
19 is 13's  right child
15 is 11's  right child
21 is 15's   left child

== 合并ha和hb后的详细信息: 
10 is root
11 is 10's   left child
12 is 11's   left child
15 is 12's   left child
21 is 15's   left child
24 is 12's  right child
36 is 24's   left child
13 is 11's  right child
17 is 13's   left child
23 is 17's   left child
19 is 13's  right child
16 is 10's  right child
20 is 16's   left child
30 is 20's   left child
40 is 30's   left child