DS树--带权路径和

DS树–带权路径和

---

题目描述

计算一棵二叉树的带权路径总和，即求赫夫曼树的带权路径和。

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

---

输出样例

在这里插入代码片

---

代码

#include<iostream>
using namespace std;
class BiTreeNode {
public:
    char data;
    BiTreeNode *LeftChild;
    BiTreeNode *RightChild;
    BiTreeNode *Parent;
    int Length;
    BiTreeNode():LeftChild(NULL),RightChild(NULL),Parent(NULL),Length(0){}
};

class BiTree {
public:
    void CreateTree(string s);
    void SetLength(int *a);
private:
    int pos;
    string strTree;
    int FindParent(BiTreeNode *t,int sum);
    void SetLength(BiTreeNode *t, int *a);
    BiTreeNode *CreateTree();
    BiTreeNode *Root;
};

int total = 0;


int BiTree::FindParent(BiTreeNode *t,int sum)
{

    if(t->Parent!=Root&&t->Parent)
    {
        sum++;
        sum = FindParent(t->Parent,sum);
        return sum;
    }
    return sum;
}

int pos2=0;

void BiTree::SetLength(int *a) { SetLength(Root,a); }

void BiTree::SetLength(BiTreeNode *t, int *a) {
    if(t)
    {
        SetLength(t->LeftChild,a);
        SetLength(t->RightChild,a);
        if(t->LeftChild==NULL&&t->RightChild==NULL)
        {
            int k = FindParent(t,1);
            t->Length = a[pos2++]*k;
            total += t->Length;
        }
    }
}

void BiTree::CreateTree(string s) {
    pos = 0;
    strTree.assign(s);
    Root = CreateTree();
}

BiTreeNode *BiTree::CreateTree() {
    BiTreeNode *T;
    char ch;
    ch = strTree[pos++];
    if(ch=='0')
        T = NULL;
    else
    {

        T = new BiTreeNode();
        T->data = ch;
        T->LeftChild = CreateTree();
        if(T->LeftChild)
             T->LeftChild->Parent = T;
        T->RightChild = CreateTree();
        if(T->RightChild)
            T->RightChild->Parent = T;
    }
    return T;
}

int main()
{
    int t;
    cin>>t;
    for(int i=0;i<t;i++)
    {
        string s;
        cin>>s;
        BiTree biTree;
        biTree.CreateTree(s);
        int n;
        cin>>n;
        int a[n];
        for(int j=0;j<n;j++)
            cin>>a[j];
        pos2=0;
        total = 0;
        biTree.SetLength(a);
        cout<<total<<endl;
    }
    return 0;
}