Huffman树的创建

来源:互联网 发布:js正则表达式可以为空 编辑:程序博客网 时间:2024/05/17 09:06

运用了小堆堆的基本操作

#include<stdio.h>#include<iostream>using namespace std;#include"heap.cpp"//引用堆的创建文件template<class T>struct Node{    Node(T weight,T data = T())        :_pLeft(NULL)        ,_pRight(NULL)        ,_weight(weight)        ,_data(data)    {}    Node* _pLeft;    Node* _pRight;    T _data;    T _weight;};template<class T>struct ComplareNode{    bool operator()(Node<T>* pLeft,Node<T>* pRight)    {        return pLeft->_weight<pRight->_weight;    }};template<class T>class HuffmanTree{public:    HuffmanTree()        :_pRoot(NULL)    {}    HuffmanTree(const T array[],size_t size)    {        //首先,我们要将数组中所有权值的数转化为节点的森林        //然后选择权值最小的两个节点做为左右子树组建成一棵树,根节点的权值为这两个节点的权值之和,将该树放入我们的森林里        //再继续重复上一步骤,直到剩下最后一个节点,该节点就是一棵huffman树的根节点        //针对于如何在森林中选取权值最小的两个节点,我们可以根据我们刚刚学的堆来处理,用森另中的所有节点建立一个小堆,每次选取堆顶的元素        _CreateHuffmanTree(array,size);    }private:    void _CreateHuffmanTree(const T array[],size_t size)    {        heap<Node<T>*,ComplareNode<T>> hp;//创建一个堆,用来存放森林        for(size_t idx=0; idx<size; ++idx)//将权值转化为节点存入堆中        {            hp.Insert(new Node<T>(array[idx]));        }        //存在问题!!!!我们前面实现的堆在调整时比较的是节点的data值        //因为我们的vector里存放的是T类型,但是我们现在vector里存放的是Node<T>*,所以调整时不能再用 <和>直接比较        //如果直接比较比较的将是两个地址,而我们要比较的则是节点的权值,但vector里又不能存放权值,如果存放的是权值,        //新产生的树将会丢失,苏所以我们必须重新定义比较类        while(hp.Size()>1)        {            Node<T>* pLeft = hp.Top();//在堆中选取两个最小权值的节点,作为左右孩子组建成新的树            hp.Remove();            Node<T>* pRight = hp.Top();            hp.Remove();            Node<T>* pParent = new Node<T>(pLeft->_weight+pRight->_weight);            pParent->_pLeft = pLeft;            pParent->_pRight = pRight;            hp.Insert(pParent);//将树放入森林        }        _pRoot = hp.Top();    }private:    Node<T>* _pRoot;};
0 0
原创粉丝点击