Huffman树的创建

运用了小堆堆的基本操作

#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;
};