在代码中有其他需求可自行按备注更改,本代码是学校实验任务,参照课本及网上资料以及自己更改编写完成。
已知a,b,c,d四个字符权值是7,5,2,4。
(1)建立相应的哈夫曼编码树,构造哈夫曼编码表,
(2)在此基础上对压缩文件进行压缩。
输出要求:
(1)输出哈夫曼数组;字符的哈夫曼编码;
(2)文件原文;压缩后的编码。
在本代码编写中,可以输入任意字符与权值对应起来,提高了代码的实际应用性。通过原文压缩、编码的实际应用体现在发送报文中,方便编码与报文解析。不过代码中还是存在些许bug,比如输入格式有点丑等,有意交流的可以留下自己的见解。
1、Huffman.cpp文件
#define _CRT_SECURE_NO_WARNINGS 1
#include <stdio.h>
#include <iostream>
#include <sstream>
#include <string>
#include "Huffman.h"
using namespace std;
unsigned int a, b;
int m;
//inline void operator delete(void* ptr) { };
void CreateHuffmanTree(HuffmanTree &HT, int n)
{
int s1, s2, i;
if (n <= 1) return;
m = 2 * n - 1;
HT = new HTNode[m + 1]; //0号单元未用,所有需要动态分配m+1个单元,HT[m]表示根结点
for (i = 1; i <= m; ++i) //将1~m号单元中的双亲、左孩子、右孩子的下标都初始化为0
{
HT[i].parent = 0; HT[i].lchild = 0; HT[i].rchild = 0;
}
cout << "请输入字符与对应的权值:\n";
for (i = 1; i <= n; ++i) //输入各个结点名称与叶子结点的权值
{
// scanf("%c=%d", &HT[i].zifu, &HT[i].weight);
cin >> HT[i].zifu;
cout << " = ";
cin >> HT[i].weight;
}
for (i = n + 1; i <= m; ++i) //通过n-1次的选择、删除、合并来创建哈夫曼树
{
Select(HT, i - 1, &s1, &s2);
HT[s1].parent = i; //得到新结点i,从森林中删除s1,s2,将s1和s2的双亲域由0改为i
HT[s2].parent = i;
HT[i].lchild = s1; //s1,s2分别作为i的左右孩子
HT[i].rchild = s2;
HT[i].weight = HT[s1].weight + HT[s2].weight; //i的权值为左右孩子的权值之和
}
}
void CreatHuffmanCode(HuffmanTree HT, HuffmanCode& HC, int n)
{
//从叶子到根逆向求每个字符的赫夫曼编码,存储在编码表HC中
int i, start, c, f;
HC = new char* [n + 1]; // 分配n个字符编码的头指针矢量
char* cd = (char*)malloc(sizeof(char) * n); // 分配临时存放编码的动态数组空间
cd[n - 1] = '\0'; // 编码结束符
for (i = 1; i <= n; ++i)
{ // 逐个字符求赫夫曼编码
start = n - 1; // start开始时指向最后,即编码结束符位置
c = i;
f = HT[i].parent; // f指向结点c的双亲结点
while (f != 0)
{ // 从叶子结点开始向上回溯,直到根结点
--start; // 回溯一次start向前指一个位置
if (HT[f].lchild == c)
cd[start] = '0'; // 结点c是f的左孩子,则生成代码0
else
cd[start] = '1'; // 结点c是f的右孩子,则生成代码1
c = f;
f = HT[f].parent; // 继续向上回溯
}
// 求出第i个字符的编码
HC[i] = new char[n - start];// 为第i 个字符编码分配空间
strcpy(HC[i], &cd[start]); // 将求得的编码从临时空间cd复制到HC的当前行中
}
delete cd; // 释放临时空间
}
2、Huffman.h文件
#pragma once
#ifndef HUFFMAN_H
#define HUFFMAN_H
#define To(a,b) a=b
#define Equal(a,b) a==b
#define Sense(i) i==0
#define Small(i,j) i<j
#define NEqual(a,b) a!=b
#define uint unsigned int
//---------哈夫曼树的存储表示---------(最优树)
typedef struct {
char zifu;
int weight; //结点的权值
int parent, lchild, rchild; //结点的双亲、左孩子、右孩子的下标
}HTNode, * HuffmanTree; //动态分配数组存储哈夫曼树
enum { MAX = 65536 };
typedef char** HuffmanCode;
void Select(HuffmanTree HT, int* min1, int* min2);
void CreateHuffmanTree(HuffmanTree& HT, int n);
void CreatHuffmanCode(HuffmanTree HT, HuffmanCode& HC, int n);
#endif
3、main.cpp函数
#define _CRT_SECURE_NO_WARNINGS 1
#include <iostream>
#include <iomanip>
#include "Huffman.h"
using namespace std;
HuffmanTree HUF;
HuffmanCode HUF_CODE;
char yuanwen[8];
void show(HuffmanTree HT, HuffmanCode HC, int n)
{
for (int i = 1; i <= 2 * n - 1; i++)
{
cout << i << "\t" << HT[i].weight << "\t" << HT[i].parent << "\t" << HT[i].lchild << "\t" << HT[i].rchild << endl;
}
for (int i = 1; i <= n; i++)
cout << HT[i].zifu << "编码为" << HC[i] << endl;
}
void second_cin(HuffmanTree HT, HuffmanCode HC, int n)
{
cout << "请输入原文内容:";
cin >> yuanwen;
cout << "压缩后编码:";
for (int i = 0; i < 8; i++)
{
for (int y = 1; y <= n; y++)
{
if (yuanwen[i] == HT[y].zifu)
{
cout << HC[y];
}
}
}
}
int main()
{
HuffmanTree HT;
HuffmanCode HC;
int n;
cout << "请输入叶子结点的个数:";
cin >> n; //输入赫夫曼树的叶子结点个数
CreateHuffmanTree(HT, n);
CreatHuffmanCode(HT, HC, n);
show(HT, HC,n);
second_cin(HT, HC, n);
system("pause");
return 0;
}
实验结果: