每日日报

哈夫曼树哈夫曼编码

#include<iostream>
#include<cstdio>
using namespace std;
const int N=100010;
struct node{
int num,fa,ch;
};
bool st[N];
int n;
node* nodes;
int find(int n){
int p=0;
for(int i=1;i<=n;i++){
if(st[i]==0&&(p==0||nodes[i].num<nodes[p].num)){
p=i;
}
}
return p;
}
void add(int lc,int rc,int p){
nodes[lc].fa=p;
nodes[lc].ch=-1;
nodes[rc].fa=p;
nodes[rc].ch=1;
nodes[p].num=nodes[lc].num+nodes[rc].num;
}
string huffman(int k){
string s;
if(nodes[k].fa){
s=huffman(nodes[k].fa);
}
if(nodes[k].ch){
s+=nodes[k].ch==-1?"0":"1";
}
return s;
}
int main(){
cin>>n;
nodes=new node[2*n];
for(int i=1;i<=n;i++){
cin>>nodes[i].num;
}
if(n==1||n==0){
cout<<"error";
}else{
int l=n+1,r=2*n-1;
while(l<=r){
int c1=find(l-1);
st[c1]=1;
int c2=find(l-1);
st[c2]=1;
add(c1,c2,l++);
}
int wpl=0;
for(int i=1;i<=n;i++){
string s=huffman(i);
cout<<nodes[i].num<<"编码为"<<s<<endl;
wpl+=nodes[i].num*s.size();
}
cout<<"WPL:"<<wpl;
}
return 0;
}