7-4 D. DS树--二叉树之最大路径

题目描述

给定一颗二叉树的逻辑结构（先序遍历的结果，空树用字符‘0’表示，例如AB0C00D00），建立该二叉树的二叉链式存储结构

二叉树的每个结点都有一个权值，从根结点到每个叶子结点将形成一条路径，每条路径的权值等于路径上所有结点的权值和。编程求出二叉树的最大路径权值。如下图所示，共有4个叶子即有4条路径，

路径1权值=5 + 4 + 11 + 7 = 27路径2权值=5 + 4 + 11 + 2 = 22

路径3权值=5 + 8 + 13 = 26路径4权值=5 + 8 + 4 + 1 = 18

可计算出最大路径权值是27。

该树输入的先序遍历结果为ABCD00E000FG00H0I00，各结点权值为：

A-5，B-4，C-11，D-7，E-2，F-8，G-13，H-4，I-1

输入

第一行输入一个整数t，表示有t个测试数据

第二行输入一棵二叉树的先序遍历，每个结点用字母表示

第三行先输入n表示二叉树的结点数量，然后输入每个结点的权值，权值顺序与前面结点输入顺序对应

以此类推输入下一棵二叉树

输入样例：

2
AB0C00D00
4 5 3 2 6
ABCD00E000FG00H0I00
9 5 4 11 7 2 8 13 4 1

输出

每行输出每棵二叉树的最大路径权值，如果最大路径权值有重复，只输出1个

输出样例：

11
27

代码

#include <iostream>
using namespace std;

struct Node{
    char data;
    int weight;
    Node *left;
    Node *right;
};

class Tree{
public:
    Node *root;
    int maxNum=0;
    void create(Node *&t){
        char c;
        cin >> c;
        if(c == '0'){
            t = NULL;
        }else{
            t = new Node;
            t->data = c;
            create(t->left);
            create(t->right);
        }
    }

    void getWeight(Node *t){  //赋权重
        if(t == NULL){
            return;
        }else{
            if(t->data >= 'A' && t->data <= 'Z'){
                cin >> t->weight;
            }
        }
        getWeight(t->left);
        getWeight(t->right);
    }

    void ToSum(Node *t){  //将每条路径和权值累加到对应的叶子结点上
        if(t == NULL){
            return;
        }else{
            if(t->left != NULL){
                t->left->weight += t->weight;
            }
            if(t->right != NULL){
                t->right->weight += t->weight;
            }
        }
        ToSum(t->left);
        ToSum(t->right);
    }

    void getRes(Node *t){  //比较所有叶子结点的权值，此时权值max即为二叉树的最大路径
        if(t == NULL){
            return;
        }else{
            if(t->left == NULL && t->right == NULL){
                if(t->weight > maxNum){
                    maxNum = t->weight;
                }
            }
        }
        getRes(t->left);
        getRes(t->right);
    }
};

int main()
{
    int t,n;
    cin >> t;
    while(t--){
        Tree tree;
        tree.create(tree.root);
        cin >> n;
        tree.getWeight(tree.root);
        tree.ToSum(tree.root);
        tree.getRes(tree.root);
        cout << tree.maxNum << endl;
    }
    return 0;
}