剑指:根据输入的前序和中续遍历序列重建二叉树

题目描述

输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。

在二叉树的前序遍历中,第一个数字总是树的根节点,但在中序遍历中根节点在序列的中间,左子树的节点序列位于跟节点的左边,右子树的节点序列位于根节点的右边。


实现:

TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin)
    {
        int length=pre.size();
        if(length==0) return NULL;
        int rootValue=pre[0];
        //根据前序遍历的第一个值创建根节点
        TreeNode *rootNode=new TreeNode(rootValue);
        if(length==1) return rootNode;
        vector<int>::iterator iterPre=pre.begin();
        vector<int>::iterator iterIn=vin.begin();
        //获取中序序列的根节点位置
	    while(*iterIn!=rootValue) iterIn++;
        //iterIn= std::find(vin.begin(), vin.end(), pre[0]);
        //获取左右子树的遍历序列
        vector<int> vinLeft(vin.begin(),iterIn);
        vector<int> vinRight(iterIn+1,vin.end());
        int leftSubTreeLength=vinLeft.size();
        //int rightSubTreeLength=vinRight.size();
        vector<int> preLeft(pre.begin()+1,pre.begin()+1+leftSubTreeLength);
        vector<int> preRight(pre.begin()+1+leftSubTreeLength,pre.end());
        rootNode->left=reConstructBinaryTree(preLeft,vinLeft);
        rootNode->right=reConstructBinaryTree(preRight,vinRight);
        
        return rootNode;
    }


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值