数据结构实验---哈夫曼编码

在代码中有其他需求可自行按备注更改，本代码是学校实验任务，参照课本及网上资料以及自己更改编写完成。

已知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;
}

实验结果：