c++实现哈夫曼编码译码
时间: 2024-01-02 10:03:44 浏览: 147
对于C++实现哈夫曼编码和译码,可以按照以下步骤进行:
1. 定义哈夫曼树的节点结构,包括字符、权重和左右子节点等信息。
2. 构建哈夫曼树:
- 创建一个优先队列(最小堆),用于存储节点,并按照权重进行排序。
- 将每个字符作为一个节点插入到优先队列中。
- 从优先队列中取出权重最小的两个节点,合并为一个新节点,并将新节点插入到优先队列中。
- 重复上述步骤,直到优先队列中只剩下一个节点,即为哈夫曼树的根节点。
3. 生成哈夫曼编码表:
- 从根节点开始,遍历哈夫曼树的每个节点。
- 对于每个节点,记录从根节点到该节点的路径,0表示左子节点,1表示右子节点。
- 将每个字符与对应的路径组成的编码表进行关联。
4. 进行编码:
- 读取待编码的文本。
- 对于每个字符,根据编码表找到对应的哈夫曼编码。
- 将编码后的二进制序列保存起来。
5. 进行译码:
- 读取待译码的二进制序列。
- 从根节点开始,根据序列中的每个位进行遍历。
- 如果是0,移动到当前节点的左子节点;如果是1,移动到当前节点的右子节点。
- 当遍历到叶子节点时,输出对应的字符,并回到根节点。
下面是一个C++实现哈夫曼编码和译码的简单示例:
```cpp
// 定义哈夫曼树节点结构
struct Node {
char data;
int weight;
Node* left;
Node* right;
};
// 构建哈夫曼树
Node* buildHuffmanTree(vector<Node*>& nodes) {
while (nodes.size() > 1) {
sort(nodes.begin(), nodes.end(), [](Node* a, Node* b) {
return a->weight < b->weight;
});
Node* left = nodes[0];
Node* right = nodes[1];
Node* parent = new Node();
parent->weight = left->weight + right->weight;
parent->left = left;
parent->right = right;
nodes.erase(nodes.begin(), nodes.begin() + 2);
nodes.push_back(parent);
}
return nodes[0];
}
// 生成哈夫曼编码表
void generateHuffmanCode(Node* root, string code, unordered_map<char, string>& codeTable) {
if (root->left == nullptr && root->right == nullptr) {
codeTable[root->data] = code;
return;
}
generateHuffmanCode(root->left, code + "0", codeTable);
generateHuffmanCode(root->right, code + "1", codeTable);
}
// 进行编码
string encode(string text, unordered_map<char, string>& codeTable) {
string encodedText;
for (char c : text) {
encodedText += codeTable[c];
}
return encodedText;
}
// 进行译码
string decode(string encodedText, Node* root) {
string decodedText;
Node* current = root;
for (char bit : encodedText) {
if (bit == '0') {
current = current->left;
} else {
current = current->right;
}
if (current->left == nullptr && current->right == nullptr) {
decodedText += current->data;
current = root;
}
}
return decodedText;
}
// 示例用法
int main() {
string text = "hello world";
vector<Node*> nodes;
for (char c : text) {
Node* node = new Node();
node->data = c;
node->weight = count(text.begin(), text.end(), c);
node->left = nullptr;
node->right = nullptr;
nodes.push_back(node);
}
Node* root = buildHuffmanTree(nodes);
unordered_map<char, string> codeTable;
generateHuffmanCode(root, "", codeTable);
string encodedText = encode(text, codeTable);
string decodedText = decode(encodedText, root);
cout << "Encoded text: " << encodedText << endl;
cout << "Decoded text: " << decodedText << endl;
return 0;
}
```
阅读全文