DS二叉树–赫夫曼树的构建与编码(含代码框架)
题目描述
给定n个权值,根据这些权值构造huffman树,并进行huffman编码
参考课本算法,注意数组访问是从位置1开始
要求:赫夫曼的构建中,默认左孩子权值不大于右孩子权值
代码框架参考如下:
输入
第一行输入t,表示有t个测试实例
第二行先输入n,表示第1个实例有n个权值,接着输入n个权值,权值全是小于1万的正整数
依此类推
输出
逐行输出每个权值对应的编码,格式如下:权值-编码
即每行先输出1个权值,再输出一个短划线,再输出对应编码,接着下一行输入下一个权值和编码。
以此类推
样例查看模式
正常显示
查看格式
输入样例1 <-复制
1
5 15 4 4 3 2
输出样例1
15-1
4-010
4-011
3-001
2-000
#include <iostream>
#include <deque>
#include <algorithm>
#include <string>
using namespace std;
class node{
public:
int val;node*left,*right;
string str;
node(){
val=0;left= nullptr;right= nullptr;str="";
}
};
bool cmp(node *a,node*b){
if(a->val<b->val)
return true;
else
return false;
}
void pre(node *p){
if(p->left)
p->left->str=p->str+"0";
if(p->right)
p->right->str=p->str+"1";
if(p->left)pre(p->left);
if(p->right)pre(p->right);
}
int main(){
int t;cin>>t;
while (t--){
int n;cin>>n;
deque<node*>d;deque<node*>so;
for (int i = 0; i < n; ++i) {
node *s=new node;cin>>s->val;
d.push_back(s);so.push_back(s);
}
while (d.size()!=1){
sort(d.begin(),d.end(),cmp);
node *l=d.front();d.pop_front();
node *r=d.front();d.pop_front();
node *s=new node;s->left=l;s->right=r;s->val=l->val+r->val;
d.push_back(s);
}
node *root=d.front();
pre(root);
for (int i = 0; i < so.size(); ++i) {
cout<<so[i]->val<<"-"<<so[i]->str<<endl;
}
}
}
- 真神奇同一颗树传到不同容器里,一个容器改变另一个也会变。