DS树--带权路径和

题目描述

二叉树的创建使用含空树表示的先序遍历序列，计算一棵二叉树的带权路径总和，即求赫夫曼树的带权路径和。

已知一棵二叉树的叶子权值，该二叉树的带权路径和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;
}