二叉树叶子节点的霍夫曼编码

#include<iostream>
#include<list>

using namespace std;

typedef struct treenode
{
    treenode* leftchild;
    treenode* rightchild;
    int data;
}Treenode,*pTreenode;

//遍历list<int>中的元素
void printL(list<int> &l)
{
    for(list<int>::iterator it=l.begin();it!=l.end();it++)
    {
        cout<<(*it);
    }
}


//对二叉树的每个叶子节点进行霍夫曼编码，将编码的结果存储在链表l中并打印出来，打印的叶子节点的顺序按照先序遍历的顺序。编码的原则是二叉树向左是0，向右是1
void Huffmancode(pTreenode root,list<int> &l)
{
    if(!root)
    {
        return;
    }

    if(!root->leftchild && !root->rightchild)
    {
        cout<<root->data<<":";
        printL(l);
        cout<<endl;
    }

    if(root->leftchild)
    {
        l.push_back(0);//将元素"0"压入栈的末端
        Huffmancode(root->leftchild,l);
        l.pop_back();//执行完递归函数的这一层，需要将其弹出，就只剩下上一层的根节点的编码了，这位为后面该根节点的右子树叶子节点编码扫清了障碍
    }

    if(root->rightchild)
    {
        l.push_back(1);//将元素"1"压入栈的末端
        Huffmancode(root->rightchild,l);
        l.pop_back();
    }
}

 

int main()
{
    //建立一颗树
    Treenode t1,t2,t3,t4,t5,t6,t7;
    memset(&t1,0,sizeof(Treenode));
    memset(&t2,0,sizeof(Treenode));
    memset(&t3,0,sizeof(Treenode));
    memset(&t4,0,sizeof(Treenode));
    memset(&t5,0,sizeof(Treenode));
    memset(&t6,0,sizeof(Treenode));
    memset(&t7,0,sizeof(Treenode));

    t1.data=1;
    t2.data=2;
    t3.data=3;
    t4.data=4;
    t5.data=5;
    t6.data=6;
    t7.data=7;

    t1.leftchild=&t2;
    t1.rightchild=&t3;
    t2.leftchild=&t4;
    t2.rightchild=&t5;
    t3.rightchild=&t6;
    t4.leftchild=&t7;
    //霍夫曼编码
    list<int> L;
    Huffmancode(&t1,L);

    return 0;
}