PTA哈夫曼编码 分数 30 作者 陈越 单位 浙江大学

给定一段文字，如果我们统计出字母出现的频率，是可以根据哈夫曼算法给出一套编码，使得用此编码压缩原文可以得到最短的编码总长。然而哈夫曼编码并不是唯一的。例如对字符串"aaaxuaxz"，容易得到字母 ‘a’、‘x’、‘u’、‘z’ 的出现频率对应为 4、2、1、1。我们可以设计编码 {‘a’=0, ‘x’=10, ‘u’=110, ‘z’=111}，也可以用另一套 {‘a’=1, ‘x’=01, ‘u’=001, ‘z’=000}，还可以用 {‘a’=0, ‘x’=11, ‘u’=100, ‘z’=101}，三套编码都可以把原文压缩到 14 个字节。但是 {‘a’=0, ‘x’=01, ‘u’=011, ‘z’=001} 就不是哈夫曼编码，因为用这套编码压缩得到 00001011001001 后，解码的结果不唯一，“aaaxuaxz” 和 “aazuaxax” 都可以对应解码的结果。本题就请你判断任一套编码是否哈夫曼编码。

输入格式：
首先第一行给出一个正整数 N（2≤N≤63），随后第二行给出 N 个不重复的字符及其出现频率，格式如下：

c[1] f[1] c[2] f[2] … c[N] f[N]
其中c[i]是集合{‘0’ - ‘9’, ‘a’ - ‘z’, ‘A’ - ‘Z’, ‘_’}中的字符；f[i]是c[i]的出现频率，为不超过 1000 的整数。再下一行给出一个正整数 M（≤1000），随后是 M 套待检的编码。每套编码占 N 行，格式为：

c[i] code[i]
其中c[i]是第i个字符；code[i]是不超过63个’0’和’1’的非空字符串。

输出格式：
对每套待检编码，如果是正确的哈夫曼编码，就在一行中输出"Yes"，否则输出"No"。

注意：最优编码并不一定通过哈夫曼算法得到。任何能压缩到最优长度的前缀编码都应被判为正确。

输入样例：
7
A 1 B 1 C 1 D 3 E 3 F 6 G 6
4
A 00000
B 00001
C 0001
D 001
E 01
F 10
G 11
A 01010
B 01011
C 0100
D 011
E 10
F 11
G 00
A 000
B 001
C 010
D 011
E 100
F 101
G 110
A 00000
B 00001
C 0001
D 001
E 00
F 10
G 11
输出样例：
Yes
Yes
No
No
代码长度限制
16 KB
时间限制
400 ms
内存限制
64 MB

答案：

# include<stdio.h>
# include<stdlib.h>
# include<string.h>

// 冒泡排序函数，用于在数组内一定范围进行排序
void bubbleSort(int arr[], int left, int right)
{
	int flag = 1;//用于检测在一轮冒泡排序中是否进行了交换操作。
	for (int i = left; i < right && flag; i++)//如果在某轮排序中没有发生任何交换，那么数组已经有序，不需要继续排序，因此可以提前结束排序过程。
	{
		flag = 0;
		for (int j = left; j < right - (i - left); j++)//已经沉底的部分就不用继续参与循环了。
		{
			if (arr[j] > arr[j + 1])
			{
				flag = 1;
				int temp = arr[j];
				arr[j] = arr[j + 1];
				arr[j + 1] = temp;
			}
		}
	}
}

int main()
{
	int n;// 输入的字符频率数
	char character;//输入的字符
	int huffNum = 0;// Huffman编码的总频率
	int frequencies[63] = { 0 };// 各字符的频率
	int huffQueue[63] = { 0 };// Huffman编码队列
	int front = 0, rear = 0;
	scanf("%d", &n);

	// 读取字符和频率
	for (int i = 0; i < n; i++)
	{
		getchar();//跳过输入缓冲区中的换行符（\n）或其他空白字符，以便能够读取下一个字符或字符串。
		scanf("%c %d", &character, &huffQueue[rear]);//因为字符是什么对算法没什么影响，所以不必全部存下，只需要保证频率全部输入即可。
		frequencies[rear] = huffQueue[rear];
		rear++;
	}

	// 对 Huffman 编码队列进行排序
	bubbleSort(huffQueue, front, n - 1);

	// 构建 Huffman 树，求哈夫曼的树的权值。
	while (front + 1 != rear)//这里千万不要写错！！！！
	{
		int node1 = huffQueue[front];
		huffQueue[front] = 0;
		int node2 = huffQueue[++front];
		huffQueue[front] = node1 + node2;
		huffNum += node1 + node2;//深度为n的结点相当于连加n-1次（n-1即其路径长度）,相当于权值*路径长
		bubbleSort(huffQueue, front, n - 1);
	}

	int m;// 查询的次数
	scanf("%d", &m);

	for (int k = 0; k < m; k++)
	{
		char codes[63][63] = { 0 };// 存储字符的 Huffman 编码
		int observedHuffNum = 0;// 观察到的 Huffman 编码的总频率

		// 读取字符和对应的 Huffman 编码
		for (int i = 0; i < n; i++)
		{
			getchar();
			scanf("%c %s", &character, codes[i]);
			observedHuffNum += (int)strlen(codes[i]) * frequencies[i];//路径长*频率
		}
		if (observedHuffNum != huffNum)printf("No\n");
		else
		{
			int flag = 1;
			for (int i = 0; i < n; i++)
			{
				for (int j = i + 1; j < n; j++)
				{
					// 判断是否存在前缀相等的情况
					if (strlen(codes[i]) <= strlen(codes[j]) && strncmp(codes[i], codes[j], strlen(codes[i])) == 0)
					{
						flag = 0;
						break;
					}
					else if (strlen(codes[i]) > strlen(codes[j]) && strncmp(codes[i], codes[j], strlen(codes[j])) == 0)
					{
						flag = 0;
						break;
					}
				}
				if (!flag)break;
			}
			if (!flag)printf("No\n");
			else printf("Yes\n");
		}
	}
	return 0;
}

如果想在VS2022里运行，需要将所有scanf换成scanf_s
并且将第39行scanf("%c %d", &character, &huffQueue[rear]);
改成scanf_s("%c %d", &character, 1, &huffQueue[rear], 1);
将第70行scanf("%c %s", &character, codes[i]);
改成scanf_s("%c %s", &character, 1, codes[i], 63);