DS树--带权路径和

题解：

1. 这里str2用来存各叶子结点的权值，其的大小应该设为LeafNum，但是LeafNum的值在编译阶段并不确定，C++中不允许将变量作为数组大小，所以不能直接声明int str2[LeafNum]，而应该先声明为int *str2，等输入LeafNum后再给str2申请LeafNum大小大的空间。因此当需要定义以变量为长度的结构时，应该通过指针加new的方式定义。
2. 这里面PreOrder函数中的TotalRoad传引用而Level不传引用很好地展现了递归函数的特性，TotalRoad在不同层之间是连续的，因此要传引用，而Level在每一层中独立，每层递归中的值都不同，当从内层递归回到外层递归后，Level的值会恢复原先这层Level的值。

题目：

问题 I: DS树--带权路径和
时间限制: 1 Sec  内存限制: 128 MB
提交: 388  解决: 328
[提交][状态][讨论版]
题目描述
计算一棵二叉树的带权路径总和，即求赫夫曼树的带权路径和。

已知一棵二叉树的叶子权值，该二叉树的带权案路径和APL等于叶子权值乘于根节点到叶子的分支数，然后求总和。如下图中，叶子都用大写字母表示，权值对应为：A-7，B-6，C-2，D-3

树的带权路径和 = 7*1 + 6*2 + 2*3 + 3*3 = 34

本题二叉树的创建参考前面的方法

输入
第一行输入一个整数t，表示有t个二叉树

第二行输入一棵二叉树的先序遍历结果，空树用字符‘0’表示，注意输入全是英文字母和0，其中大写字母表示叶子

第三行先输入n表示有n个叶子，接着输入n个数据表示n个叶子的权值，权值的顺序和前面输入的大写字母顺序对应

以此类推输入下一棵二叉树

输出
输出每一棵二叉树的带权路径和

样例输入
2
xA00tB00zC00D00
4 7 6 2 3
ab0C00D00
2 10 20
样例输出
34
40

代码块：

#include <iostream>
using namespace std;

class HTNode
{
private:
    char data;
    int weight;
    HTNode *lchild, *rchild;
public:
    HTNode():weight(0), lchild(NULL), rchild(NULL){}
    ~HTNode(){}
    friend class HuffmanTree;
};

class HuffmanTree
{
private:
    HTNode *root;
    int LeafNum;
    string str1;
    int *str2;
public:
    HuffmanTree();
    ~HuffmanTree();
    void CreateTree();
    void CreateTree(HTNode *&p, int &i, int &j);
    void PreOrder();
    void PreOrder(HTNode *p, int Level, int &TotalRoad);
};

HuffmanTree::HuffmanTree()
{
    cin>>str1;
    cin>>LeafNum;
    str2 = new int[LeafNum];
    for(int i=0; i<LeafNum; i++)
        cin>>str2[i];
}

HuffmanTree::~HuffmanTree()
{
    delete root;
    delete []str2;
}

void HuffmanTree::CreateTree(HTNode *&p, int &i, int &j)
{
    if(str1[i]!='0')
    {
        p = new HTNode;
        if(str1[i]>=65 && str1[i]<=90)
        {
            p->data = str1[i];
            p->weight = str2[j];
            j++;
        }
        i++;
        CreateTree(p->lchild, i, j);
        i++;
        CreateTree(p->rchild, i, j);
    }
    else
        p = NULL;
}

void HuffmanTree::CreateTree()
{
    int i = 0;
    int j = 0;
    CreateTree(root, i, j);
}

void HuffmanTree::PreOrder()
{
    int Level = 0;
    int TotalRoad = 0;
    PreOrder(root, Level, TotalRoad);
    cout<<TotalRoad<<endl;
}

void HuffmanTree::PreOrder(HTNode *p, int Level, int &TotalRoad)
{
    if(p)
    {
        if(!p->lchild && !p->rchild)
        {
            TotalRoad += p->weight*Level;
        }
        Level++;
        PreOrder(p->lchild, Level, TotalRoad);
        PreOrder(p->rchild, Level, TotalRoad);
    }
}

int main(void)
{
    int t;
    cin>>t;
    while(t--)
    {
        HuffmanTree myTree;
        myTree.CreateTree();
        myTree.PreOrder();
    }
    return 0;
}