JZ7 重建二叉树

描述
给定节点数为 n 的二叉树的前序遍历和中序遍历结果,请重建出该二叉树并返回它的头结点。
例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序{4,7,2,1,5,3,8,6},则重建出如下图所示。
在这里插入图片描述

输入:
[1,2,4,7,3,5,6,8],[4,7,2,1,5,3,8,6]
返回值:
{1,2,3,4,#,5,6,#,7,#,#,8}
说明:
返回根节点,系统会输出整颗二叉树对比结果,重建结果如题面图示    

思路:知道前序遍历和中序遍历,那么在前序遍历中知道根节点之后,能在中序遍历中划分出左子树和右子树,分别对左子树和右子树递归进行这一过程即可
在这里插入图片描述

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
  public:
    map<int, int>map;
    //map中存放中序遍历元素的下标
    TreeNode* reConstructBinaryTree(vector<int> pre, vector<int> vin) {
        //判断是否为空
        if (pre.size() == 0)
            return nullptr;
        for (int i = 0; i < vin.size(); i++)
            map[vin[i]] = i;
        TreeNode* root = f(pre, 0, pre.size() - 1, vin, 0, vin.size() - 1);
        return root;
    }
    TreeNode* f(vector<int>pre, int l1, int r1, vector<int>vin, int l2, int r2) {
        if (l1 > r1 && l2 > r2)
            return nullptr;
        TreeNode* root = new TreeNode(pre[l1]);
        //取出当前前序遍历中的根节点在中序遍历中的下标
        int i = map[pre[l1]];
        //进行划分
        root->left = f(pre, l1 + 1, l1 + i - l2, vin, l2, i - 1);
        root->right = f(pre, l1 + i - l2 + 1, r1, vin, i + 1, r2);
        return root;

    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值