分治法 72、73、根据前序遍历和中序遍历或者根据中序遍历和后序遍历创建树

根据前序遍历和中序遍历树构造二叉树.

描述

根据前序遍历和中序遍历树构造二叉树.

你可以假设树中不存在相同数值的节点

您在真实的面试中是否遇到过这个题?   是

样例

给出中序遍历:[1,2,3]和前序遍历:[2,1,3]. 返回如下的树:

  2
 / \
1   3
class Solution {
public:
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        return construct(pre,0,pre.size()-1,vin,0,vin.size()-1);
    }
    TreeNode* construct(vector<int>&pre,int pleft,int pright,vector<int>&vin,int vleft,int vright)
    {
        if(pleft>pright||vleft>vright) return NULL;
        TreeNode* cur=new TreeNode(pre[pleft]);
        int i=0;
        for(i=vleft;i<vin.size();i++)
        {
            if(cur->val==vin[i])
                break;
        }
        cur->left=construct(pre,pleft+1,i+pleft-vleft,vin,vleft,i-1);
        cur->right=construct(pre,i+pleft-vleft+1,pright,vin,i+1,vright);
        return cur;
    }
};

根据中序遍历和后序遍历重建二叉树

样例

给出树的中序遍历: [1,2,3] 和后序遍历: [1,3,2]

返回如下的树:

  2

 /  \

1    3

递归解法(分治法)

分析:
这道题要求从中序和后序遍历的结果来重建原二叉树,我们知道中序的遍历顺序是左-根-右,后序的顺序是左-右-根,对于这种树的重建一般都是采用递归来做,可参见我之前的一篇博客Convert Sorted Array to Binary Search Tree 将有序数组转为二叉搜索树。针对这道题,由于后序的顺序的最后一个肯定是根,所以原二叉树的根节点可以知道,题目中给了一个很关键的条件就是树中没有相同元素,有了这个条件我们就可以在中序遍历中也定位出根节点的位置,并以根节点的位置将中序遍历拆分为左右两个部分,分别对其递归调用原函数。代码如下: 

class Solution {
public:
    /**
     * @param inorder: A list of integers that inorder traversal of a tree
     * @param postorder: A list of integers that postorder traversal of a tree
     * @return: Root of a tree
     */
    TreeNode * buildTree(vector<int> &inorder, vector<int> &postorder) {
        // write your code here
        return  construct(inorder,0,inorder.size()-1,postorder,0,postorder.size()-1);
    }
 TreeNode* construct(vector<int>&inorder,int ileft,int iright,vector<int>&postorder,int pleft,int pright)
        {
            if(ileft>iright||pleft>pright)
            return NULL;
            TreeNode* cur=new TreeNode(postorder[pright]);
            int i=0;
            for(i=ileft;i<inorder.size();i++)
            {
                if(inorder[i]==cur->val)
                break;
            }
            cur->left=construct(inorder,ileft,i-1,postorder,pleft,pleft+i-1-ileft);
            cur->right=construct(inorder,i+1,iright,postorder,pleft+i-ileft,pright-1);
            return cur;
        }
};



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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值