文件压缩——哈夫曼树编码（一）

 1 #pragma once
 2 
 3 #include<queue>
 4 
 5 template<typename W>
 6 struct HuffmanTreeNode
 7 {
 8     W val;
 9     HuffmanTreeNode* right;
10     HuffmanTreeNode* left;
11 };
12 
13 //堆——优先级队列。priority_queue
14 
15 template<typename W,typename T>
16 class HuffmanTree
17 {
18     //放在内部，方便且安全
19     typedef HuffmanTreeNode Node;
20 public:
21     //优先队列中放的是结点的指针。
22     //因为我放的是指针，但是比较的时候不能比较指针啊。
23     //所以在这里我需要写一个仿函数。
24     //让他用我的仿函数比较。但是要注意，仿函数是第三个模板参数，所以第二个模板参数还是要一样的。
25 
26     //因为我们要杜绝出现出现次数为0的情况，这种情况不会被我们统计但是如果不管会占空间。
27     HuffmanTree(W* a,size_t n,const T& invalid);
28 
29     //生成完哈夫曼树后那么接下来就是要生成哈夫曼编码了
30     //生成哈夫曼编码的方法
31     //1.三叉链法。添加一个parents指针在HuffmanNode结构体中，并且注意初始化。
32     //2.递归，左0右1。递归方法放在FileCompress类中。
33 
34     //因为如果利用递归法，就很有可能需要在最后，将根的哈夫曼码变成当前结点结构体中的哈夫曼码。
35     //所以需要类提供一个接口
36     //返回root
37     Node* GetRoot();
38 
39     ~HuffmanTree();
40     //因为涉及树，所以删除很有可能需要调用递归，要自觉习惯用一个函数来完成析构任务。
41     void destory(Node* root);
42 private:
43     //暂时用不到拷贝构造和=运算符重载
44     //所以直接用防拷贝的方法。
45     HuffmanTree(const HuffmanTree<W> &t);
46     HuffmanTree<W>& operator=(const HuffmanTree<T> &t);
47 protected:
48     Node* root;
49 };

 1 #pragma once
 2 
 3 #include<string>
 4 #include"HuffmanTree.h"
 5 using namespace std;
 6 
 7 //一个字符的信息——
 8 //我们会将这个值统计好传过去来生成哈夫曼树。
 9 //但同时我们也知道哈夫曼树中有W的加的操作，>的操作，!=的操作。
10 //所以我们还需要重载+运算符，>运算符，!=运算符。
11 struct CharInfo
12 {
13     char _ch;
14     long long _count;
15     string _code;
16     CharInfo operator+(const CharInfo& info);
17     bool operator>(const CharInfo& info);
18     bool operator!=(const CharInfo& info);
19 };
20 
21 class FileCompress
22 {
23 public:
24     //构造函数就是写死，初始化哈希表。
25     FileCompress();
26     //压缩
27     //1.首先要统计文件中字符出现的次数。
28     //2.用IO流的方式读字符串---因为是读，所以是ifstream，因为要从文件里读。
29     //C++reference里有些，构造函数就直接讲字符串当做参数(字符串实际为一个文件)，读字符就用get。
30     void Compress(const char *file);
31     //解压缩
32     void Uncompress(const char *file);
33     void GenerateHuffmanCode(HuffmanTreeNode<char> *node);
34 protected:
35     CharInfo _hashInfos[256];//用来存放各个字符。
36 };