C语言数据结构作业之哈夫曼树

#include <iostream>
#include <fstream>
#include <string.h>
using namespace std; 

#define MaxSize 1024  
#define OK 1
#define ERROR 0
typedef int Status;

typedef struct wordcnt{  
    char ch;
    int cnt = 0;
}Count;

typedef struct NumCount{  
    Count count[MaxSize];
    int length = 0;
}NumCount;

typedef struct HTree{  
    char data;
    int weight;
    int parent,lchild,rchild;
}HTNode,*HuffmanTree; 

typedef struct HCode{ 
    char data;
    char* str; 
}*HuffmanCode;

Status ReadData(char *source);  
Status WordCount(char *data,NumCount *paraCnt);  
Status Show(NumCount *paraCnt);    
Status CreateHuffmanTree(HuffmanTree &HT,int length,NumCount cntarray);   
Status select(HuffmanTree HT,int top,int *s1,int *s2);   
Status CreateHuffmanCode(HuffmanTree HT,HuffmanCode &HC,int length);  
Status Encode(char *data,HuffmanCode HC,int length);  
Status Decode(HuffmanTree HT,int length);   

int main(int argc, char** argv) {
    char data[MaxSize];  
    NumCount Cntarray;
    ReadData(data);   
    WordCount(data,&Cntarray);   
 
    HuffmanTree tree;
    CreateHuffmanTree(tree,Cntarray.length,Cntarray);  
    HuffmanCode code;  
    CreateHuffmanCode(tree,code,Cntarray.length);  
    Encode(data,code,Cntarray.length);  
    Decode(tree,Cntarray.length);  
    cout<<"Please view the generated TXT file to check the result"<<endl; 
    return 0;
}

Status ReadData(char *source)
{
     
    ifstream infile;
    infile.open("in.txt");
    cout<<"Reading..."<<endl;
    cout<<"the input file is:"<<endl;
    infile.getline(source,MaxSize);
    cout<<source<<endl;
    infile.close();
    cout<<endl;
    return OK;
}

Status WordCount(char *data,NumCount *paraCnt)
{
    int flag; 
    int len = strlen(data);
    for(int i = 0;i < len;++i)
    {
        flag = 0;
        for(int j = 0;j < paraCnt->length;++j)
        {
            if(paraCnt->count[j].ch == data[i]) // 若已有记录，直接++ 
            {
                ++paraCnt->count[j].cnt;
                flag = 1;
                break;
            }
            
        }
        if(!flag)  
        {
            paraCnt->count[paraCnt->length].ch = data[i];
            ++paraCnt->count[paraCnt->length].cnt;
            ++paraCnt->length;
        }
    }
    return OK;
}

Status Show(NumCount *paraCnt)
{
    cout<<"the length is "<<paraCnt->length<<endl;
    for(int i = 0;i < paraCnt->length;++i)
    {
        cout<<"The character "<<paraCnt->count[i].ch<<"  appears  "<<paraCnt->count[i].cnt<<endl;
    }
    cout<<endl;
    return OK;
}

Status CreateHuffmanTree(HuffmanTree &HT,int length,NumCount cntarray)
{
    if(length <= 1) return ERROR;
    int s1,s2;
    int m = length*2-1;  
    HT = new HTNode[m+1];
    for(int i = 1;i <= m;++i)  
    {
        HT[i].parent = 0;
        HT[i].lchild = 0;
        HT[i].rchild = 0;
    }
    
    for(int i = 1;i <= length;++i) 
    {
        HT[i].data = cntarray.count[i-1].ch;
        HT[i].weight = cntarray.count[i-1].cnt;
    }
    
    for(int i = length + 1;i <= m;++i)
    {
        select(HT,i-1,&s1,&s2);   
        HT[s1].parent = i;
        HT[s2].parent = i;
        HT[i].lchild = s1;
        HT[i].rchild = s2;
        HT[i].weight = HT[s1].weight + HT[s2].weight;   
    }
    return OK;
}

Status select(HuffmanTree HT,int top,int *s1,int *s2)
{
    int min = INT_MAX;
    for(int i = 1;i <= top;++i)   
    {
        if(HT[i].weight < min && HT[i].parent == 0)
        {
            min = HT[i].weight;
            *s1 = i;
        }
    }
    
    min = INT_MAX;
    for(int i = 1;i <= top;++i)   
    {
        if(HT[i].weight < min && i != *s1 && HT[i].parent == 0)
        {
            min = HT[i].weight;
            *s2 = i;
        }
    }
    return OK;    
}

Status CreateHuffmanCode(HuffmanTree HT,HuffmanCode &HC,int length)
{
    HC = new HCode[length+1];
    char *cd = new char[length];  
    cd[length-1] = '\0';   
    int c,f,start;
    for(int i = 1;i <= length;++i)
    {
        start = length-1;  // 
        c = i;
        f = HT[c].parent;
        while(f != 0)
        {
            --start;   
            if(HT[f].lchild == c)
                cd[start] = '0';
            else 
                cd[start] = '1';
            c = f;
            f = HT[c].parent;
        }
        HC[i].str = new char[length-start];  
        HC[i].data = HT[i].data;
        strcpy(HC[i].str,&cd[start]);   
    }
    delete cd;
}

Status Encode(char *data,HuffmanCode HC,int length)
{
    ofstream outfile;
    outfile.open("code.txt");
    for(int i = 0;i < strlen(data);++i)   
    {
        for(int j = 1;j <= length;++j)
        {
            if(data[i] == HC[j].data)
            {
                outfile<<HC[j].str;
            }
        }
    }
    outfile.close();
    cout<<"the code txt has been written"<<endl;
    cout<<endl;
    return OK;
}

Status Decode(HuffmanTree HT,int length)
{
    char codetxt[MaxSize*length];
    ifstream infile;
    infile.open("code.txt");
    infile.getline(codetxt,MaxSize*length);
    infile.close();
    
    ofstream outfile;
       outfile.open("out.txt");
    
    int root = 2*length-1;  // 从根节点开始遍历 
    for(int i = 0;i < strlen(codetxt);++i)
    {
        if(codetxt[i] == '0') root = HT[root].lchild;  //为0表示向左遍历 
        else if(codetxt[i] == '1') root = HT[root].rchild; //为1表示向右遍历 
        if(HT[root].lchild == 0 && HT[root].rchild == 0)  // 如果已经是叶子节点，输出到输出文件中，然后重新回到根节点 
        {
            outfile<<HT[root].data;
            root = 2*length-1;
        }
    }
    outfile.close();
    cout<<"the output txt has been written"<<endl;
    cout<<endl;
    return OK;
}
————————————————

                            版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
                        
原文链接：https://blog.csdn.net/gz040725/article/details/138870346