wpl计算方法_C++哈夫曼树WPL的计算 求代码

展开全部

#include

using namespace std;

struct node

{

int w;

int flag;

int lchild,rchild;

int parent;

};

node huff[1001];

int n;

void read()

{

int i;

cout<

cin>>n;

for(i=1;i<=n;i++)

cin>>huff[i].w;

}

int huffman()

{

int i,min1,min2,k1,k2,j,s;//初始化标记

for(i=1;i<=2*n-1;i++)

{

huff[i].flag=0;

huff[i].lchild=-1;

huff[i].rchild=-1;

huff[i].parent=-1;

}

//n-1次建树

for(i=1;i<=n-1;i++)

{

min1=100000;

min2=100000;//求最小初始化最大

k1=-1;// 记下最小次小的下标

k2=-1;

for(j=1;j<=n+i-1;j++)//搜索范围

{

if((huff[j].w

{

min2=min1;//最小给次小

k2=k1;//记下次小下标

min1=huff[j].w;//把最小赋值给min1

k1=j; //记下最小下标

}

else if((huff[j].w

{

min2=huff[j].w;//直接修改次小值

k2=j;//记下次小的下标

}

}

huff[n+i].w=min1+min2;//新增加的结点

huff[k1].flag=1;//填最小与次小的标记：

huff[k2].flag=1;

huff[k1].parent=n+i;//填最小与次小的parent：

huff[k2].parent=n+i;

huff[n+i].lchild=k1;//填新增结点的左右孩子:

huff[n+i].rchild=k2;

}

s=0;//WPL的值为非叶子结点之和

for(i=n+1;i<=2*n-1;i++)

s=s+huff[i].w;

cout<

return s;

}

void print(int t,string s1)

{

if ((huff[t].lchild==-1)&&(huff[t].rchild==-1))

cout<

else

{

print(huff[t].lchild,s1+"0");

print(huff[t].rchild,s1+"1");

}

}

int main()

{

read();

huffman();

print(2*n-1,"");

}