题目描述
二叉树的创建使用含空树表示的先序遍历序列,计算一棵二叉树的带权路径总和,即求赫夫曼树的带权路径和。
已知一棵二叉树的叶子权值,该二叉树的带权路径和WPL等于叶子权值乘于根节点到叶子的分支数,然后求总和。
如下图中,叶子都用大写字母表示,权值对应为: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>
#include <map>
using namespace std;
#define MAX (int)1e5
string s;
int i, ans;
pair<int, int> *tree;
map<char, int> m;
void create(int n, int l) {
i++;
if (i > s.size() || s[i] == '0') return;
tree[n].first = i;
tree[n].second = l;
create(n * 2 + 1, l + 1);
create(n * 2 + 2, l + 1);
}
void preTraval(int n) {
if (n >= MAX || tree[n].first == -1) return;
if (s[tree[n].first] >= 'A' && s[tree[n].first] <= 'Z')
ans += m[s[tree[n].first]] * tree[n].second;
preTraval(n * 2 + 1);
preTraval(n * 2 + 2);
}
int main() {
int t;
cin >> t;
tree = new pair<int, int>[MAX];
while (t--) {
s.clear();
m.clear();
ans = 0;
for (int ii = 0; ii < MAX; ii++) {
tree[ii].first = -1;
tree[ii].second = 0;
}
i = -1;
cin >> s;
create(0, 0);
int n;
cin >> n;
int index = 0;
while (n--) {
while (s[index] < 'A' || s[index] > 'Z') {
index++;
}
cin >> m[s[index++]];
}
preTraval(0);
cout << ans << endl;
}
return 0;
}