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表示二叉树的结点数量，然后输入每个结点的权值，权值顺序与前面结点输入顺序对应

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

输出

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

样例输入

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

样例输出

11
27

代码

#include "bits/stdc++.h"
using namespace std;
const int maxn=1e5+20;
struct NODE{
    char data;
    int val;
    NODE *l,*r;
    NODE(){
        l=NULL;
        r=NULL;
    }
};
string s;
int id,n,t,ans;
map<char,int> mp;

void build(NODE *&head){
    if(id>=n || s[id]=='0'){
        head=NULL;
        id++;
        return;
    }
    head=new NODE;
    head->data=s[id];
    id++;
    build(head->l);
    build(head->r);
}

void find(NODE *head){
    if(head->l==NULL && head->r==NULL){
//        cout<<head->val<<endl;
        ans=max(ans,head->val);
        return ;
    }
    if(head->l){
        head->l->val=head->val+mp[head->l->data];
        find(head->l);
    }
    if(head->r){
        head->r->val=head->val+mp[head->r->data];
        find(head->r);
    }
}

int main(){
//    freopen("123.in","r",stdin);
    cin>>t;
    while(t--){
        id=ans=0;
        mp.clear();
        cin>>s;
        n=s.length();
        NODE *head;
        build(head);
        int m;
        cin>>m;
        for(int i=0;i<n;i++){
            if(!s[i] || s[i]=='0') continue;
            int x;
            cin>>x;
            mp[s[i]]=x;
//            cout<<s[i]<<"--"<<x<<endl;
        }
        head->val=mp[head->data];
//        cout<<head->data<<" "<<head->val<<endl;
        find(head);
        cout<<ans<<endl;
    }
    return 0;
}