题解:
- 这里str2用来存各叶子结点的权值,其的大小应该设为LeafNum,但是LeafNum的值在编译阶段并不确定,C++中不允许将变量作为数组大小,所以不能直接声明int str2[LeafNum],而应该先声明为int *str2,等输入LeafNum后再给str2申请LeafNum大小大的空间。因此当需要定义以变量为长度的结构时,应该通过指针加new的方式定义。
- 这里面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
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/df2829754ef18bbcf9f14f1fa082c71f.png)
本题二叉树的创建参考前面的方法
输入
第一行输入一个整数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;
}