剑值offer66题之每日一题——第四题

题目描述:

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

思路:运用递归的思想,每一棵子树的构建方式都一样,先递归的构建左右子树,最后生成大树返回根节点

代码实现:

class Solution {
public:
    TreeNode* reConstruct(vector<int> pre,//原始先序数列
                          int startPre,//当前子树先序在原始先序数列中的起始位置
                          int endPre,//当前子树先序在原始先序数列中的结束位置
                          vector<int> vin,//原始中序数列
                          int startVin,//当前子树中序在原始中序数列中的起始位置
                           int endVin )//当前子树中序在原始中序数列中的结束位置
    {
        if(startPre>endPre||startVin>endVin) return NULL;
        TreeNode* root=new TreeNode(pre[startPre]);//当前先序的第一个数为当前子树的根节点
    
        for(int i=startVin;i<=endVin;i++)//在当前子树的中序数列中找到根节点下标位置i
        {
            if(vin[i]==pre[startPre])
            {
                //先构建当前子树的左子树
                root->left=reConstruct(pre,startPre+1,startPre+i-startVin,vin,startVin,i-1);
                //在构建当前子树的右子树
                root->right=reConstruct(pre,startPre+i-startVin+1,endPre,vin,i+1,endVin);
                break;
            }
        }
        //当前树的左右子树构建完成,返回这棵树
        return root;        
    }
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
      int Prelength=pre.size();//先序长度
      int Vinlength=vin.size();//中序长度
      return reConstruct(pre,0,Prelength-1,vin,0,Vinlength-1);
    }
};

猜你喜欢

转载自blog.csdn.net/qq_41598851/article/details/82945808