一、Huffman编码
Huffuman编码的过程主要分为两步,第一步根据字符出现的权值构建Huffuman树,第二步,遍历Huffuman树找出对应字符的编码
1. 构建Huffuman树代码如下:
typedef struct iNode { int weight; int parent, lchild, rchild; } huffman_tree, *phuffman_tree;
int sum_weight(int *weight, int len)//计算权值之和 { int sum = 0; for(int i = 0; i < len; i++) sum += weight[i]; return sum; }
void select(huffman_tree *ht, int len, int &s1, int &s2)//搜索ht数组中parent为0的最小两个节点,分别存入s1,s2,len为要搜索的长度
{ s1 = s2 = 0; for(int i = 1; i <= len; i++) { if(ht[i].parent == 0) { if(ht[s1].weight >= ht[i].weight) { s1 = i; s2 = s1; } } } }
void make_tree(int *weight, huffman_tree* ht, int len)//weight是字符的权重,ht是存储二叉树的节点数组 { int m = 2*len - 1; int s1 = 0, s2 = 0; ht[0] = {sun_weight(weight, len) + 1, 0, 0, 0};//0号节点权设置为总权值加1便于select时设置比较初值 for(int i = 1; i <= len; i++) ht[i] = {weight[i - 1], 0, 0, 0}; for( ; i <= m; i++) ht[i] = {0, 0, 0, 0}; for(i = n + 1; i <= m; i++) { select(ht, i - 1, s1, s2); ht[s1].parent = i; ht[s2].parent = i; ht[i].lchild = s1; ht[i].rchild = s2; ht[i].weight = ht[s1].weight + ht[i].weight; } }
1. 构建Huffuman编码
这里有两种构建方法,第一种是由叶子到根的构建,代码如下:
vodi make_code_from_leaf(huffman_tree *ht, int len, huffman_code *hc)//从叶子结点到根的编码过程 { char code[len]; //int start = len; //code[len - 1] = "\0"; for (int i = 1; i <= len; i++) { int start = len; code[--start] = "\0"; int next = i; while (ht[next].parent ! = 0) { if (ht[ht[next].parent].lchild = next) code[--start] = "0"; else code[--start] = "1"; next = ht[next].parent; } strncpy(hc[i], &code[start], len - start); } }
从根节点开始遍历求huffuman编码的代码如下(注意weight这里发挥标记的作用:
void make_code_from_root(huffman_tree *ht, int len, huffman_code *hc) { char code[len]; int clen = 0; int p = 2*len - 1; for ( int i = 1; i <= p ; i++) ht[i].weight = 0; while (p) { if(ht[p].weight == 0) {//左节点遍历 ht[p].weight = 1; if(ht[p].lchild != 0) { p = ht[p].lchild; code[clen++] = "0"; } else if(ht[p].rchild == 0) { code[clen++] = "\0"; hc[p] = (char *)malloc(clen + 1); strcpy([hc[p], code); } } else if (ht[p].weight == 1) {//右子树遍历 ht[p].weight = 2; if (ht[p].rchild != 0) { code[clen++] = "1"; p = ht[p].rchild; } } else { //到达叶子结点回溯另一分支 ht[p].weight = 0; p = ht[p].parent; clen--; } } }
void make_code_from_root(huffman_tree *ht, int len, huffman_code *hc) { char code[len]; int clen = 0; int p = 2*len - 1; for ( int i = 1; i <= p ; i++) ht[i].weight = 0; while (p) { if(ht[p].weight == 0) { ht[p].weight = 1; if(ht[p].lchild != 0) { p = ht[p].lchild; code[clen++] = "0"; } else if(ht[p].rchild == 0) { code[clen++] = "\0"; hc[p] = (char *)malloc(clen + 1); strcpy([hc[p], code); } } else if (ht[p].weight == 1) { ht[p].weight = 2; if (ht[p].rchild != 0) { code[clen++] = "1"; p = ht[p].rchild; } } else { ht[p].weight = 0; p = ht[p].parent; clen--; } } }
void make_code_from_root(huffman_tree *ht, int len, huffman_code *hc) { char code[len]; int clen = 0; int p = 2*len - 1; for ( int i = 1; i <= p ; i++) ht[i].weight = 0; while (p) { if(ht[p].weight == 0) { ht[p].weight = 1; if(ht[p].lchild != 0) { p = ht[p].lchild; code[clen++] = "0"; } else if(ht[p].rchild == 0) { code[clen++] = "\0"; hc[p] = (char *)malloc(clen + 1); strcpy([hc[p], code); } } else if (ht[p].weight == 1) { ht[p].weight = 2; if (ht[p].rchild != 0) { code[clen++] = "1"; p = ht[p].rchild; } } else { ht[p].weight = 0; p = ht[p].parent; clen--; } } }
一、Huffman译码
根据Huffuman编码规则的规则可知,如果所有左子树代表“0”,右子树代表“1”则,可以用搜索码中的“0”来作为字符的分割码,但是要注意最后的全1码的识别其代码如下:
int select_hc(huffman_code *hc, huffman_code s_hc, int len)//根据获得的编码与Huffuman编码表进行比较获取该码字在码表中的位置 { for (int i = 0; i < len; i++) if (strcmp(hc[i], s_hc) == 0) return i; return -1; } /*******************encode function********************** ********************input : code is 0 an 1 serial************************* ********************hc is huffman code table, chars is the encoded characters, hc_len is char sum, len is code length****/ void decode(char *code, huffuman_code *hc, char *chars, char *encode_char, int hc_len, int len)// { char temp_code[10]; for (int i = 0, j = 0, k = 0; i < len; i++) { if (code[i] == "0") { temp_code[k++] = "0"; temp_code[k++] = "\0"; int select_pos = select_hc(hc, temp_code, hc_len); if (select_pps >= 0) encode_char[j++] = chars[select_pos]; else printf("error code : %s invalid", temp_code); k = 0; } else {
temp_code[k++] = "1";
if (strcmp(temp_code, hc[hc_len - 1]) == 0) {//注意处理“1”
temp_code[k++] = "\0";
encode_char[j++] = chars[hc_len - 1];
k = 0;
}
} } }