题目描述
给定n个权值,根据这些权值构造huffman树,并进行huffman编码参考课本算法,注意数组访问是从位置1开始
要求:赫夫曼的构建中,默认左孩子权值不大于右孩子权值
输入 第一行输入t,表示有t个测试实例 第二行先输入n,表示第1个实例有n个权值,接着输入n个权值,权值全是小于1万的正整数 依此类推
输出 逐行输出每个权值对应的编码,格式如下:权值-编码
即每行先输出1个权值,再输出一个短划线,再输出对应编码,接着下一行输入下一个权值和编码。 以此类推样例输入
1
5 15 4 4 3 2
样例输出
15-1
4-010
4-011
3-001
2-000
#include <iostream>
#include <string>
using namespace std;
class HuffNode
{
public:
unsigned int weight;
unsigned int parent, Lchild, Rchild;
};
class HuffManTree
{
private:
void MakeTree();
void SelectMin(int pos, int *s1, int *s2);
public:
int len; //结点数量
int lnum; //叶子数量
HuffNode *hufftree;
string * huffCode; //每个字符对应Huffman编码
void MakeTree(int n, int wt[]);
void Coding();
~HuffManTree();
};
void HuffManTree::MakeTree(int n, int wt[])
{
int i;
lnum = n;
len = 2*n - 1;
hufftree = new HuffNode[2*n];
huffCode = new string[lnum+1]; //位置从1开始计算
//huffCode实质是个二维字符数组,第i行表示第i个字符对应的编码
//赫夫曼树初始化 -- 叶子结点部分
for(i=1; i<=n; i++)
{
hufftree[i].weight = wt[i-1]; //填入叶子权值,位置从1开始
hufftree[i].parent = 0;
hufftree[i].Lchild = 0;
hufftree[i].Rchild = 0;
} //end for
//赫夫曼树初始化 -- 非叶子结点部分
for(i=n+1; i<=len; i++)
{
hufftree[i].weight = 0;
hufftree[i].parent = 0;
hufftree[i].Lchild = 0;
hufftree[i].Rchild = 0;
} //end for
MakeTree();
}
void HuffManTree::MakeTree()
{
int i, s1, s2;
//赫夫曼数构建
for(i=lnum+1; i<=len; i++)
{
SelectMin(i-1, &s1, &s2);
hufftree[s1].parent = i;
hufftree[s2].parent = i;
hufftree[i].Lchild = s1;
hufftree[i].Rchild = s2;
hufftree[i].weight = hufftree[s1