c++哈夫曼树的文件压缩解压程序全部代码及设计报告.docVIP

#include iostream #include fstream #include queue //队列容器 using namespace std; const int leaf = 256; //最多可能出现的不同字符数 const long MAX = //表示无穷大 typedef struct HTnode { long weight; //记录结点的权值 int parent; //记录结点的双亲结点位置 int lchild; //结点的左孩子 int rchild; //结点的右孩子 int *code; //记录该结点的huffman编码 int codelen; //记录该结点huffman编码的长度 HTnode() { weight = MAX; parent = -1; lchild = -1; rchild = -1; codelen = 0; } }HTnode; class huffmanTree { public: huffmanTree(); virtual ~huffmanTree(); bool count(char *input); //统计各字符出现的次数，将其写入对应结点的权值 void create(); //构造huffman树 void code(); //计算每个字符的huffman编码 void addbit(int bit); //压缩时对一个未满8个bit的byte中加入一个bit void resetbyte(); //将byte清空 bool compress(char *input, char *output); //压缩函数 成功执行返回 true 失败 false bool decompress(char *input, char *output); //解压函数 成功执行返回 true 失败 false void compare(char *input, char *output); //将原文件与压缩后的文件比较 private: int root; //记录根结点的位置 int leafnum; //记录不同字符的个数 HTnode HT[leaf*2-1]; //HTnode结构的数组，用来表示huffman树，树的最大结点个数不会超过leaf*2-1 char byte; //压缩文件时用来缓冲bit的变量 int bitsnum; //byte中bit的个数 int lacknum; //压缩到最后byte中的bit不满8个时填充的0的个数 }; huffmanTree::huffmanTree() { //初始化成员变量 root = 0; leafnum = 0; byte = 0; bitsnum = 0; lacknum = 0; } huffmanTree::~huffmanTree() { for(int i=0; ileaf; i++) { if(HT[i].codelen != 0) delete []HT[i].code; } } //统计各字符出现的次数 bool huffmanTree::count(char *input) { ifstream ifs; char c; ifs.open(input,ios::binary); if(!ifs){ cout 无法打开文件 input ! endl; return false; } while(ifs.get(c)){ if(HT[c+128].weight==MAX){ //若该字符是第一次出现，先初始化权值 HT[c+128].weight = 0; leafnum++; } HT[c+128].weight++; //权值+1 } ifs.close(); return true; } //选权值最小的两棵树组成新的数 void huffmanTree::create() { for(int i=leaf; i2*leaf-1; i++) { int loc1=-1, loc2=-1; for(int j=0; ji; j++){ if(HT[j].parent != -1) continue; if(loc1==-1 || HT[j].weight HT[loc1].weight) { lo