【数据结构】树的应用-计算哈夫曼树的WPL值

计算哈夫曼树的WPL值        

根据给定的n个权值（非负值），计算所构造哈夫曼树的WPL值。

基本要求：    

（1）根据给定的数据，建立哈夫曼树；      

（2）输出每个叶子结点的带权路径长度；      

（3）输出哈夫曼树的WPL值。

测试数据要求：        

输入的n个权值之和应为100，且不允许有负值。 

---

核心代码：
 

void createhuffman(int &t,int n){
    int i;
    for(i=1;i<=n;i++){
        f[i].data=a[i].data;
        f[i].lchild=f[i].rchild=0;
        a[i].addr=i;
    }
    t=n+1;
    i=n;
    while(i>=2){
        sort(a+1,a+i+1);
        f[t].data=a[1].data+a[2].data;
        f[t].lchild=a[1].addr;
        f[t].rchild=a[2].addr;
        a[1].data=f[t].data;
        a[1].addr=t;
        a[2].data=a[i].data;
        a[2].addr=a[i].addr;
        i--;t++;
    }
}

---

完整的调试代码：

#include
#include
#include
#include
#define N 100
using namespace std;
typedef struct Node{
    int data,addr;
    bool operator < (const Node &p) const {
        return data=2){
        sort(a+1,a+i+1);
        f[t].data=a[1].data+a[2].data;
        f[t].lchild=a[1].addr;
        f[t].rchild=a[2].addr;
        a[1].data=f[t].data;
        a[1].addr=t;
        a[2].data=a[i].data;
        a[2].addr=a[i].addr;
        i--;t++;
    }
}
void dfs(int root,int L=0){
    if(Check_leaf(root)){
        printf("结点 %d:  路径长度为 L = %d \n",f[root].data,L);
        printf("带权路径长度为: %d * %d = %d\n\n",f[root].data,L,f[root].data*L);
        ans+=f[root].data*L;
        return ;    
    }
    dfs(f[root].lchild,L+1);
    dfs(f[root].rchild,L+1);
}
int main()
{
    int n,root,t;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i].data);
    }
    createhuffman(root,n);
    root--;
    dfs(root);
    printf("树的带权路径长度(WPL): %d\n",ans);
}
/*
4
2 3 7 9

//ans=38

6
9 12 6 3 5 15

//ans=122
*/