105.从前序与中序遍历序列构造二叉树

在这里插入图片描述
前序遍历:中左右

中序遍历:左中右

例:
 pre:[3,9,20,15,7]  in[9,3,15,20,7];

前序遍历的第一个数3,就是根节点,而中序遍历3所在的位置左边,包含在该根节点的左子树里,右边包含在右子树里,这里使用:左前,右前,左中,右中,四个区间,来表示每一子树所在的位置,如此就可以递归的构建出二叉树;

我们需要知道一个根节点的左子树和右子树的边界在前序以及中序序列中的下标,用pl, pr, il, ir来表示前序左边界、前序有边界、中序左边界、中序右边界。

图例

在这里插入图片描述

假设这一段前序遍历第一个数是x,找到中序遍历序列中x对应在中序遍历的下标k,中序遍历下标k左边直到il, 就是左子树在中序遍历的下标区间: [il, k - 1];同理,右子树下标区间:  [k + 1, ir]
在这里插入图片描述
接下来寻找左子树的在前序遍历的区间下标,设点为N,由左子树区间在两个序列中长度相等有:
N - pl = k - il
N = pl + k - il

由此递归创建即可。

代码

在开始时需要使用一个hash表来装inorder的值与对应的下标,如此来建立前序点与中序点的联系。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
unordered_map<int,int>hash;
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        for (int i = 0; i < inorder.size(); i++) hash[inorder[i]] = i;
        return build(preorder, inorder, 0, preorder.size() - 1, 0, inorder.size() - 1);
    }

    TreeNode* build(vector<int>& preorder, vector<int>& inorder, int pl, int pr, int il, int ir) {
        if (pl > pr) return NULL;
        int k = hash[preorder[pl]];
        TreeNode* root = new TreeNode(preorder[pl]);
        root->left = build(preorder, inorder, pl + 1, pl + k - il, il, k - 1);
        root->right = build(preorder, inorder, pl + k - il + 1, pr,k + 1, ir);
        return root;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值