数据结构学习_树（2）

 Huffman树是二叉树的一种应用。Huffman树的总的带权路径长度最短。

b

6

c

d

7

5

2

4

11

18

下面是自己写的构造Huffman树的代码。

随着世界越来越开放，我想自己写的代码以后可能要和国外的人交流。因此在以后的代码编写过程中注释尽量用英语。

#include <iostream>
#include <vector>
#include <algorithm>
#include <string>
using namespace std;

//the structure of the Huffman tree's nodes
typedef struct HFNode
...{
    int data;
    struct HFNode* pLChild;
    struct HFNode* pRChild;
}HFNode;

HFNode EmptyNode =
...{
    0,
    NULL,
    NULL
};

//creat the rule for the sort() to compare all the elements in the vector
bool SortRule(HFNode& a, HFNode& b)
...{
    if (a.data <= b.data)
        return true;
    else
        return false;
}

//use the preprocessor  operator ## to creat the variables
#define pHFNode(j) pHFNode##j

//print all the elements of the HFTree according pre order
void PreOrderPrint(HFNode* pRoot)
...{
    if (NULL != pRoot)
        cout << pRoot->data << endl;
    if (NULL != pRoot->pLChild)
        PreOrderPrint(pRoot->pLChild);
    else
        cout << "@" << endl;
    if (NULL != pRoot->pRChild)
        PreOrderPrint(pRoot->pRChild);
    else
        cout << "@" << endl;;
}
int main()
...{    
    const int NODENUM = 4;
                     vector<HFNode> HFArray (NODENUM, EmptyNode);
    int ValueArray[NODENUM] = ...{7, 5, 2, 4};

    //provide data for all the nodes
    for (int i = 1; i <= NODENUM; i++)
    ...{
        HFArray[i - 1].data = ValueArray[i - 1];
    }
    int a, b, c ;
    a = b = c = 1;
    while(HFArray.size() > 1)
    ...{
        //sort the vector and find the first and second of the smallest nodes
        sort(HFArray.begin(), HFArray.end(), SortRule);
        HFNode* pHFNode(a) = new HFNode;
        memcpy(pHFNode(a), &HFArray[0], sizeof(HFNode));
        a++;
        HFNode* pHFNode(b) = new HFNode;
        memcpy(pHFNode(b), &HFArray[1], sizeof(HFNode));
        b++;

        //creat the node as a root for the selected two leaf nodes 
        HFNode* pHFNode(c) = new HFNode;
        pHFNode(c)->data = pHFNode(a)->data + pHFNode(b)->data;
        pHFNode(c)->pLChild = pHFNode(a);
        pHFNode(c)->pRChild = pHFNode(b);
        c++;

        //erase the two nodes and push back the new root
        HFArray.erase(HFArray.begin(), HFArray.begin() + 2);
        HFArray.push_back(*pHFNode(c));
    }
    HFNode Root = HFArray.front();
    PreOrderPrint(&Root);
    getchar();
}