我的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++解题报告: