19_5

最新推荐文章于 2021-01-15 08:20:57 发布

啥也不会的炼丹师

最新推荐文章于 2021-01-15 08:20:57 发布

阅读量186

点赞数

本文链接：https://blog.csdn.net/weixin_43370733/article/details/104622399

版权

题目描述：

给出一个二叉排序树的层次遍历（或者给出一个二叉树的中序遍历和层次遍历），求从它的一个叶子节点到另外一个叶子节点的路径，要求路径上经过节点的数值之和最大。

分析：

题目为二叉排序树，那么按二叉排序树的键值，从小到大排序，可以得到中序序列，按照中序序列和层次遍历建树，之后进行DFS搜索。

样例输入：

7
1 2 3 4 5 6 7
4 2 6 1 3 5 7

样例输出

AC代码

#include <stdio.h>
#include <vector>
#include <algorithm>
#define maxn 100
using namespace std;


int In[maxn];
vector<int>layer;

typedef struct Node
{
    int data;
    Node*left,*right,*parent;
}Node;

Node* Create(vector<int>layer,int l,int r)
{
    if(layer.size()==0)return NULL;
    Node* root=new Node;
    root->data=layer[0];
    int k;
    for(k=l;k<=r&&In[k]!=layer[0];k++){}//
    vector<int>leftlayer,rightlayer;
    for(int i=1;i<(int)layer.size();i++)
    {
        bool isleft=false;
        for(int j=l;j<k;j++)
        {
            if(layer[i]==In[j])
            {
                isleft=true;
                break;
            }
        }
        if(isleft)leftlayer.push_back(layer[i]);
        else rightlayer.push_back(layer[i]);
    }
    root->left=Create(leftlayer,l,k-1);
    if(root->left!=NULL)root->left->parent=root;
    root->right=Create(rightlayer,k+1,r);
    if(root->right!=NULL)root->right->parent=root;
    return root;
}


int opt=0;
int update(Node* root)
{
    if(root==NULL)return 0;//
    int val1=root->data+update(root->left);//左子树+根
    int val2=root->data+update(root->right);//右子树+根
    int val3=root->data+update(root->left)+update(root->right);//左右子树+根
    if(val3>opt)opt=val3;//因为是判断根节点的距离 所以只能用第三种情况判断
    return max(val1,val2);//返回第一二种情况的最大值
}
 

int main()
{
    freopen("input.txt","r",stdin);
    //freopen("output.txt","w",stdout);
    int n;
    scanf("%d",&n);
    for(int i=0;i<n;i++)
    {
        scanf("%d",In+i);
    }
    int x;
    for(int i=0;i<n;i++)
    {
        scanf("%d",&x);
        layer.push_back(x);
    }
    Node* root=Create(layer,0,n-1);
    update(root);
    printf("%d",opt);
}