DS树--带权路径和

DS树–带权路径和

题目描述

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

已知一棵二叉树的叶子权值，该二叉树的带权路径和WPL等于叶子权值乘于根节点到叶子的分支数，然后求总和。

如下图中，叶子都用大写字母表示，权值对应为：A-7，B-6，C-2，D-3

树的带权路径和 = 71 + 62 + 23 + 33 = 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 <stack>
#include <vector>
#include <queue>
#include <list>

using namespace std;

struct BiTreeNode {
    char data;
    BiTreeNode *leftChild, *rightChild, *parents;
};

string sTree;
int pos;
int *num;
int Index;

BiTreeNode *BTree_create_tree() {
    if (sTree[pos] == '0') {
        pos++;
        return nullptr;
    } else {
        auto *node = new BiTreeNode;
        node->data = sTree[pos];
        pos++;
        node->leftChild = BTree_create_tree();
        if (node->leftChild != nullptr)
            node->leftChild->parents = node;
        node->rightChild = BTree_create_tree();
        if (node->rightChild != nullptr)
            node->rightChild->parents = node;
        return node;
    }
}

BiTreeNode *Build() {
    cin >> sTree;
    pos = 0;
    return BTree_create_tree();
}

int sum;
BiTreeNode *Root;

int level(char result) {
    int s = 0;
    if (Root == nullptr)
        return 0;
    else {
        queue<BiTreeNode *> q;
        q.push(Root);
        while (!q.empty()) {
            s++;
            queue<BiTreeNode *> r;
            while (!q.empty()) {
                r.push(q.front());
                q.pop();
            }
            while (!r.empty()) {
                BiTreeNode *node = r.front();
                if (node->data == result) {
                    s -= 1;
                    return s;
                }
                r.pop();
                if (node->leftChild != nullptr)
                    q.push(node->leftChild);
                if (node->rightChild != nullptr)
                    q.push(node->rightChild);
            }
        }
    }
}

void level(BiTreeNode *root) {
    if (root == nullptr)
        return;
    else {
        if (root->leftChild == nullptr && root->rightChild == nullptr) {
            sum = sum + level(root->data) * num[Index++];
        }
        if (root->leftChild != nullptr) {
            level(root->leftChild);
        }
        if (root->rightChild != nullptr) {
            level(root->rightChild);
        }
    }
}

int main() {
    int T;
    cin >> T;
    while (T--) {
        Root = Build();
        int size;
        cin >> size;
        sum = 0, Index = 0;
        num = new int[size];
        for (int i = 0; i < size; ++i) {
            cin >> num[i];
        }
        level(Root);
        cout << sum << endl;
    }
    return 0;
}