Data Structures and Algorithms7-9——Huffman Codes

我的Data Structures and Algorithms代码仓：https://github.com/617076674/Data-Structures-and-Algorithms

原题链接：https://pintia.cn/problem-sets/16/problems/671

题目描述：

知识点：哈夫曼编码

思路：哈夫曼编码的2个条件

（1）把每个字符的出现次数作为叶子节点的权值，求一棵树，使得这棵树的戴荃路径长度最小。

按下述操作可以构建一棵哈夫曼树：

a：初始状态下共有n个节点（节点的权值分别是给定的n个数），将它们视作n棵只有一个节点的数。

b：合并其中根节点权值最小的两棵树，生成两棵树根节点的父节点，权值为这两个根节点的权值之和，这样树的数量就少了一个。

c：重复操作b，直到只剩下一棵树为止，这棵树就是哈夫曼树。

（2）任何一个字符的编码都不是另一个字符的编码的前缀。

时间复杂度是O(M * N ^ 2)。空间复杂度是O(N)。

C++代码：

#include<iostream>
#include<queue>
#include<cstring>
#include<algorithm>

using namespace std;

bool isPre(char* str1, char* str2);	//判断str1是否是str2的前缀或str2是否是str1的前缀 

int main() {
	int N;
	scanf("%d", &N);
	int frequency[63];
	fill(frequency, frequency + 63, 0);
	char str[2];
	int count;
	priority_queue<int, vector<int>, greater<int> > q;
	for(int i = 0; i < N; i++) {
		scanf("%s %d", str, &count);
		if(str[0] >= '0' && str[0] <= '9') {
			frequency[str[0] - '0'] = count;
		} else if(str[0] >= 'a' && str[0] <= 'z') {
			frequency[str[0] - 'a' + 10] = count;
		} else if(str[0] >= 'A' && str[0] <= 'Z') {
			frequency[str[0] - 'A' + 36] = count;
		} else if(str[0] == '_') {
			frequency[62] = count;
		}
		q.push(count);
	}
	int x, y, minLen = 0;
	while(q.size() > 1) {
		x = q.top();
		q.pop();
		y = q.top();
		q.pop();
		q.push(x + y);
		minLen += x + y;
	}
	int M, len, num;
	scanf("%d", &M);
	char code[N][64];
	for(int i = 0; i < M; i++) {
		len = 0;
		for(int j = 0; j < N; j++) {
			scanf("%s %s", str, code[j]);
			if(str[0] >= '0' && str[0] <= '9') {
				len += frequency[str[0] - '0'] * strlen(code[j]);
			} else if(str[0] >= 'a' && str[0] <= 'z') {
				len += frequency[str[0] - 'a' + 10] * strlen(code[j]);
			} else if(str[0] >= 'A' && str[0] <= 'Z') {
				len += frequency[str[0] - 'A' + 36] * strlen(code[j]);
			} else if(str[0] == '_') {
				len += frequency[62] * strlen(code[j]);
			}
		}
		bool flag = true;
		for(int j = 0; j < N; j++){
			for(int k = j + 1; k < N; k++){
				if(isPre(code[j], code[k])){
					flag = false;
					break;
				}
			}
			if(!flag){
				break;
			}
		} 
		if(len == minLen && flag) {
			printf("Yes\n");
		} else {
			printf("No\n");
		}
	}
	return 0;
}

bool isPre(char* str1, char* str2) {
	int commonLen = min(strlen(str1), strlen(str2));
	for(int t = 0; t < commonLen; t++) {
		if(str1[t] != str2[t]) {
			return false;
		}
	}
	return true;
}

C++解题报告：