leetcode07:根据先序遍历和中序遍历来重建二叉树

举例:

 

思路:(1)通过先序遍历找到根节点的位置 3  再通过3在中序遍历中的位置找到3的左子树和右子树

            (2) 在3的左子树中,找到左子树的根节点(9 在先序遍历中找到),然后重复步骤(1),即在中序遍历中找根节点9的左子树和右子树

          (3)在3的右子树中,找到右子树的根节点(20 在先序遍历中 从3往右移左子树的长度就得到3的右子树根节点20),然后重复步骤(1)

代码如下:

 unordered_map<int, int> inoderIdex;
class Solution {
    //root 表示在前序遍历中的root的位置
    // left 表示 前序遍历中root节点 在中序遍历中 对应的左子树边界
    //right 表示 前序遍历中root节点 在中序遍历中 对应的右子树边界
    TreeNode * dfs(vector<int>& preorder, vector<int>& inorder, int root, int left, int right)
    {
        if(left > right) return NULL;
        int rootIdxInoder = inoderIdex[preorder[root]];
        int leftTreeSize = rootIdxInoder - left;//左子树的长度
        TreeNode * t = new TreeNode(preorder[root]);
        t ->left = dfs(preorder, inorder, root + 1, left, rootIdxInoder-1 );
        t ->right = dfs(preorder, inorder, root + leftTreeSize +1, rootIdxInoder+1, right);
        return t;
    }
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        
        if(inorder.size() == 0) return NULL;
        for(int i = 0; i < inorder.size(); i++)
        {
            inoderIdex[inorder[i]] = i;//将中序遍历的结果存储起来 便于找到左子树与右子树
        }
        return dfs(preorder, inorder, 0, 0 , inorder.size()-1);
    }
};

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值