大三总结 .

输入一组整型权值，构建哈夫曼树，实现哈夫曼编码，并输出带权路径长度。

include

include

using namespace std;
typedef char HuffmanCode;
//哈夫曼树：parent、lchild、rchild、weight
typedef struct
{
int weight;
int lchild,rchild,parent;
}HTNode,*HuffmanTree;
void min(HuffmanTree HT,int pos,int &s1,int &s2)
{//找到最小权值的两个下标信息
s1=pos;s2=pos;//这个位置parent肯定为0
for(int i=pos-1;i>=1;i--){
if(HT[i].parent0&&HT[i].weight<=HT[s1].weight)s1=i;//放个等于是想取下标小的后往前找
}
for(int i=pos-1;i>=1;i--)
if(HT[i].parent0&&HT[i].weight<=HT[s2].weight&&i!=s1)s2=i;
}
void CreatHT(HuffmanTree &HT,int n)
{//创建Huffman树
if(n<=1)return;//返回**
int m=2n-1;//总共2n-1个结点信息
HT=new HTNode[m+1];//0号位不用所以+1new
for(int i=1;i<=m;++i)//这里要注意是<=m而不是<=n
{
HT[i].weight=0;HT[i].parent=0;HT[i].lchild=0;HT[i].rchild=0;//所有设置为0
}
for(int i=1;i<=n;i++)//注意是<=n，因为1-n个结点的weight是知道的
cin>>HT[i].weight;
for(int i=n+1;i<=m;i++)
{
int s1,s2;//最小权值的下标信息s1,s2
min(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[s2].weight;//不要忘记权值了
}
}
void CreatHC(HuffmanTree HT,HuffmanCode &HC,int n)
{//创建哈夫曼编码:从结点-->根
HC= new char[n+1];//1-n所以n+1个
char cd=new char[n];
cd[n-1]='\0';//编码结束符
for(int i=1;i<=n;i++)//前面的就是结点
{ int start=n-1;
int c=i,f=HT[i].parent;
while(f!=0)//f=0说明是根结点了
{
if(c==HT[f].lchild)cd[--start]='0';
else cd[--start]='1';
c=f;f=HT[c].parent;
}
HC[i]=new char[n-start];
strcpy(HC[i],&cd[start]);//***
}
delete cd;
}
int main()
{//2n-1个结点
HuffmanTree HT;
HuffmanCode HC;
int n,WPL=0,j;
cin>>n;
if(n>1){//**加上这个
CreatHT(HT,n);
CreatHC(HT,HC,n);
for(int i=1;i<=n;i++)
{
cout<<HT[i].weight<<"编码为";
for(j=0;HC[i][j]!='\0';j++)
cout<<HC[i][j];
WPL+=j*HT[i].weight;
cout<<endl;
}
cout<<"WPL:"<<WPL;
}
else{
cout<<"error";
}
return 0;
}